Commit 3da186d7 authored by fenglei.tian's avatar fenglei.tian

add gpu mnist_mlp_forward test

parent 178de84e
./build/test/unit-test --gtest_filter=GPU.ab ./build/test/unit-test --gtest_filter=GPU.ab
./build/test/unit-test --gtest_filter=benchmark.mxnet_mnist_mlp_forward
./build/test/unit-test --gtest_filter=benchmark.gpu_mxnet_mnist_mlp_forward
./build/test/unit-test --gtest_filter=GPU.abc ./build/test/unit-test --gtest_filter=GPU.abc
./build/test/unit-test --gtest_filter=GPU.maximum ./build/test/unit-test --gtest_filter=GPU.maximum
./build/test/unit-test --gtest_filter=GPU.minimum ./build/test/unit-test --gtest_filter=GPU.minimum
......
...@@ -123,6 +123,12 @@ TEST(benchmark, mxnet_mnist_mlp_forward) ...@@ -123,6 +123,12 @@ TEST(benchmark, mxnet_mnist_mlp_forward)
run_benchmark(json_path, "CPU", 1000); run_benchmark(json_path, "CPU", 1000);
} }
TEST(benchmark, gpu_mxnet_mnist_mlp_forward)
{
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/mnist_mlp_forward.json");
run_benchmark(json_path, "GPU", 1000);
}
TEST(benchmark, mxnet_10_bucket_lstm) TEST(benchmark, mxnet_10_bucket_lstm)
{ {
const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/10_bucket_LSTM.json"); const string json_path = file_util::path_join(SERIALIZED_ZOO, "mxnet/10_bucket_LSTM.json");
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment