From bb2423a8daa1b8a55d074bfa26e643a409839dc6 Mon Sep 17 00:00:00 2001 From: zhanghonggeng Date: Thu, 13 Feb 2025 06:32:22 +0000 Subject: [PATCH] fix OpType check --- paddle2onnx/mapper/tensor/elementwise.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle2onnx/mapper/tensor/elementwise.cc b/paddle2onnx/mapper/tensor/elementwise.cc index 28f07820d..929bab558 100755 --- a/paddle2onnx/mapper/tensor/elementwise.cc +++ b/paddle2onnx/mapper/tensor/elementwise.cc @@ -34,7 +34,8 @@ REGISTER_PIR_MAPPER(elementwise_mod, ElementWiseModMapper) REGISTER_PIR_MAPPER(elementwise_floordiv, ElementWiseFloordivMapper) int32_t ElementwiseMapper::GetMinOpsetVersion(bool verbose) { - if (OpType() == "elementwise_min" || OpType() == "elementwise_max") { + if (convert_pir_op_name(OpType()) == "elementwise_min" || + convert_pir_op_name(OpType()) == "elementwise_max") { Logger(verbose, 8) << RequireOpset(8) << std::endl; return 8; }