From 99cbb648bf463ac3bc91bfb34f2ab1ab6570bfdf Mon Sep 17 00:00:00 2001 From: Minghao Liu Date: Mon, 11 May 2020 08:45:55 +0000 Subject: [PATCH] fix --- src/operator/tensor/elemwise_binary_scalar_op.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/operator/tensor/elemwise_binary_scalar_op.h b/src/operator/tensor/elemwise_binary_scalar_op.h index d87867d173db..40fef8d337de 100644 --- a/src/operator/tensor/elemwise_binary_scalar_op.h +++ b/src/operator/tensor/elemwise_binary_scalar_op.h @@ -226,7 +226,7 @@ class BinaryScalarOp : public UnaryOp { public: template static void Compute_(const nnvm::NodeAttrs &attrs, - mshadow::Stream* s, + mshadow::Stream* s, const std::vector &inputs, const std::vector &req, const std::vector &outputs) { @@ -235,7 +235,7 @@ class BinaryScalarOp : public UnaryOp { using namespace mshadow; using namespace mshadow::expr; const double alpha = nnvm::get(attrs.parsed); - MSHADOW_REAL_TYPE_SWITCH(outputs[0].type_flag_, DType, { + MSHADOW_TYPE_SWITCH(outputs[0].type_flag_, DType, { MXNET_ASSIGN_REQ_SWITCH(req[0], Req, { mxnet_op::Kernel, cpu>::Launch( s, inputs[0].Size(), outputs[0].dptr(), inputs[0].dptr(), DType(alpha)); @@ -262,7 +262,7 @@ class BinaryScalarOp : public UnaryOp { using namespace mshadow; using namespace mshadow::expr; TBlob temp_tblob; - MSHADOW_REAL_TYPE_SWITCH(outputs[0].type_flag_, DType, { + MSHADOW_TYPE_SWITCH(outputs[0].type_flag_, DType, { if (common::is_int(inputs[0].type_flag_)) { Tensor temp_tensor = ctx.requested[0].get_space_typed(Shape1(inputs[0].Size()), s);