Fix

Author: reminisce

src/operator/numpy/np_elemwise_broadcast_logic_op.cc

@@ -259,14 +259,14 @@ static constexpr char func_less_equal_scalar_cpu[] = "less_equal_scalar_cpu";
 static constexpr char func_less_equal_scalar_gpu[] = "less_equal_scalar_gpu";
 
 #if MXNET_USE_TVM_OP
-#define MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_CPU(name)                               \
-  NNVM_REGISTER_OP(_npi_##name##_scalar)                                                       \
-  .set_attr<FCompute>("FCompute<cpu>", TVMBinaryBroadcastScalarCompute{func_##name##scalar_cpu})
+#define MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_CPU(name)                                \
+  NNVM_REGISTER_OP(_npi_##name##_scalar)                                                        \
+  .set_attr<FCompute>("FCompute<cpu>", TVMBinaryBroadcastScalarCompute{func_##name##_scalar_cpu})
 
 #if MXNET_USE_CUDA
-#define MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_GPU(name)                               \
-  NNVM_REGISTER_OP(_npi_##name##_scalar)                                                       \
-  .set_attr<FCompute>("FCompute<gpu>", TVMBinaryBroadcastScalarCompute{func_##name##scalar_gpu})
+#define MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_GPU(name)                                \
+  NNVM_REGISTER_OP(_npi_##name##_scalar)                                                        \
+  .set_attr<FCompute>("FCompute<gpu>", TVMBinaryBroadcastScalarCompute{func_##name##_scalar_gpu})
 
 MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_GPU(equal);
 MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR_LOGIC_GPU(not_equal);

src/operator/tensor/elemwise_binary_broadcast_op.h

@@ -337,7 +337,8 @@ void BinaryBroadcastComputeLogic(const nnvm::NodeAttrs& attrs,
             mshadow::Shape<NDim> rstride = mxnet_op::calc_stride(new_rshape.get<NDim>());
             mxnet_op::Kernel<mxnet_op::binary_broadcast_kernel<NDim, DType, bool, OP>, xpu>::
             template LaunchEx(s, new_oshape.Size(), req[0], lstride, rstride, oshape,
-                              inputs[0].dptr<DType>(), inputs[1].dptr<DType>(), outputs[0].dptr<bool>());
+                              inputs[0].dptr<DType>(), inputs[1].dptr<DType>(),
+                              outputs[0].dptr<bool>());
           });
       });
     }

src/operator/tensor/elemwise_binary_op.h

@@ -515,7 +515,8 @@ class ElemwiseBinaryOp : public OpBase {
             if (size != 0) {
               Kernel<mxnet_op::op_with_req<OP, Req>, xpu>::Launch(s, size,
                                                                   outputs[0].dptr<bool>(),
-                                                                  inputs[0].dptr<DType>(), inputs[1].dptr<DType>());
+                                                                  inputs[0].dptr<DType>(),
+                                                                  inputs[1].dptr<DType>());
             }
         });
       });