Finally

apache · Oct 11, 2019 · da5c5c7 · da5c5c7
1 parent 5cc2615
commit da5c5c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/src/operator/numpy/np_elemwise_broadcast_logic_op.cc b/src/operator/numpy/np_elemwise_broadcast_logic_op.cc
@@ -133,16 +133,13 @@ MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC(greater_equal);
 MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC(less_equal);
 
 #if MXNET_USE_TVM_OP
+
 #define MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC_CPU(name)                          \
   NNVM_REGISTER_OP(_npi_##name)                                                    \
   .set_attr<FCompute>("FCompute<cpu>", TVMBinaryBroadcastCompute{func_##name##_cpu})
 
-#pragma message("In np_elemwise_broadcast_logic_op.cc, MXNET_USE_TVM_OP ")
-
 #if MXNET_USE_CUDA
 
-#pragma message("In np_elemwise_broadcast_logic_op.cc, MXNET_USE_CUDA")
-
 #define MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC_GPU(name)                          \
   NNVM_REGISTER_OP(_npi_##name)                                                    \
   .set_attr<FCompute>("FCompute<gpu>", TVMBinaryBroadcastCompute{func_##name##_gpu})
@@ -158,9 +155,6 @@ MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC_GPU(less_equal);
 
 #else
 
-#pragma message("In np_elemwise_broadcast_logic_op.cc, MXNET_USE_TVM_OP")
-#pragma message("In np_elemwise_broadcast_logic_op.cc, MXNET_USE_CUDA")
-
 #define MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC_CPU(name)                                     \
   NNVM_REGISTER_OP(_npi_##name)                                                               \
   .set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastComputeLogic<cpu, mshadow_op::np_##name>)

diff --git a/src/operator/numpy/np_elemwise_broadcast_logic_op.cu b/src/operator/numpy/np_elemwise_broadcast_logic_op.cu
@@ -30,11 +30,7 @@ namespace mxnet {
 namespace op {
 
 
-#if MXNET_USE_TVM_OP
-#pragma message("In np_elemwise_broadcast_logic_op.cu, MXNET_USE_TVM_OP")
-
-#else
-#pragma message("In np_elemwise_broadcast_logic_op.cu, MXNET_USE_TVM_OP")
+#if MXNET_USE_TVM_OP == 0
 
 #define MXNET_OPERATOR_REGISTER_NP_BINARY_LOGIC_GPU(name)                                     \
   NNVM_REGISTER_OP(_npi_##name)                                                               \