Fix CI

8 years ago · b4e1824363
parent 0708a1550c
commit b4e1824363
1 changed files with 2 additions and 3 deletions
--- a/paddle/operators/math/math_function.cc
+++ b/paddle/operators/math/math_function.cc
@ -272,11 +272,10 @@ struct TensorSetConstantWithPlace : public boost::static_visitor<void> {

 void set_constant(const platform::DeviceContext& context,
                  framework::Tensor* tensor, float value) {
+  TensorSetConstantWithPlace func(context, tensor, value);
 #ifdef PADDLE_WITH_CUDA
-  boost::apply_visitor(TensorSetConstantWithPlace(context, tensor, value),
-                       tensor->place());
+  tensor->place().apply_visitor(func);
 #else
-  TensorSetConstantWithPlace func(context, tensor, value);
  func(platform::CPUPlace());
 #endif
 }