update pr unbind

PaddlePaddle · May 28, 2024 · 5029e64 · 5029e64
1 parent 2012e95
commit 5029e64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/paddle2onnx/mapper/tensor/fill_constant.cc b/paddle2onnx/mapper/tensor/fill_constant.cc
@@ -29,7 +29,7 @@ int32_t FillConstantMapper::GetMinOpset(bool verbose) {
       onnx_dtype != ONNX_NAMESPACE::TensorProto::DOUBLE && 
       onnx_dtype != ONNX_NAMESPACE::TensorProto::BOOL
       ) {
-    Error() << "Only support int32/int64/float32/float64 data type in "
+    Error() << "Only support int32/int64/float32/float64/bool data type in "
                "fill_constant operator."
             << std::endl;
     return -1;
@@ -135,7 +135,7 @@ void FillConstantMapper::Opset9() {
       data[0] = static_cast<double>(value);
       auto ptr = reinterpret_cast<char*>(data.data());
       tensor->set_raw_data(std::string(ptr, sizeof(double)));
-    }else if (onnx_dtype == ONNX_NAMESPACE::TensorProto::BOOL) {
+    } else if (onnx_dtype == ONNX_NAMESPACE::TensorProto::BOOL) {
       std::vector<double> data(1);
       data[0] = static_cast<bool>(value);
       auto ptr = reinterpret_cast<char*>(data.data());

diff --git a/paddle2onnx/mapper/tensor/reduce_prod.cc b/paddle2onnx/mapper/tensor/reduce_prod.cc
@@ -24,8 +24,6 @@ int32_t ReduceMapperProd::GetMinOpset(bool verbose) {
 }
 
 void ReduceMapperProd::Opset18() {
-  //Assert(false, "ReduceMapperProd don't support opset18 now");
-#if 1
   auto axis_name_ = "dim";
   GetAttr("keep_dim", &keep_dim_);
   GetAttr("reduce_all", &reduce_all_);
@@ -65,7 +63,6 @@ void ReduceMapperProd::Opset18() {
   }
   auto out_info = GetOutput("Out");
   helper_->AutoCast(out_node_name, out_info[0].name, x_info[0].dtype, out_info[0].dtype);
-#endif
 }