Use CoreOpRunner for refactored Ops.

apache · Nov 30, 2017 · a28586f · a28586f
1 parent b161709
commit a28586f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/cpp/operator/activation_perf.cc b/tests/cpp/operator/activation_perf.cc
@@ -41,7 +41,7 @@ TEST(ACTIVATION_PERF, ExecuteBidirectional) {
  TShape shape({5, 5});
  kwargs_t kwargs = basic_activation_args;
  kwargs.push_back({"act_type", "tanh"});
- test::op::LegacyOpRunner<mxnet::op::ActivationProp, float, float> runner;
+ test::op::CoreOpRunner<mxnet::op::ActivationProp, float, float> runner;
  runner.RunBidirectional(false, { shape }, kwargs, 1);
 }
 
@@ -52,7 +52,7 @@ TEST(ACTIVATION_PERF, TimingCPU) {
  kwargs_t kwargs = basic_activation_args;
  // Which math function is arbitrary since it will have roughly constant timing among approaches
  kwargs.push_back({"act_type", "tanh"});
- test::op::LegacyOpRunner<mxnet::op::ActivationProp, float, float> runner;
+ test::op::CoreOpRunner<mxnet::op::ActivationProp, float, float> runner;
  runner.RunBidirectional(false,
  { TShape({10, 10, 10, 10}) },
  kwargs, 1); // prime code and cache

diff --git a/tests/cpp/operator/fully_conn_perf.cc b/tests/cpp/operator/fully_conn_perf.cc
@@ -41,7 +41,7 @@ const kwargs_t basic_fullyconn_args = { {"num_hidden", "250"} };
 TEST(FULLY_CONNECTED, ExecuteBidirectionalFullyConnected) {
  TShape shape({5, 5});
  kwargs_t kwargs = basic_fullyconn_args;
- test::op::LegacyOpRunner<mxnet::op::FullyConnectedProp, float, float> runner;
+ test::op::CoreOpRunner<mxnet::op::FullyConnectedProp, float, float> runner;
  runner.RunBidirectional(false, { shape }, kwargs, 1);
 }
 
@@ -50,7 +50,7 @@ TEST(FULLY_CONNECTED, ExecuteBidirectionalFullyConnected) {
  */
 TEST(FULLY_CONNECTED, FullyConnectedTimingCPU) {
  kwargs_t kwargs = basic_fullyconn_args;
- test::op::LegacyOpRunner<mxnet::op::FullyConnectedProp, float, float> runner;
+ test::op::CoreOpRunner<mxnet::op::FullyConnectedProp, float, float> runner;
  runner.RunBidirectional(false,
  { TShape({10, 10, 10, 10}) },
  kwargs, 1); // prime code and cache