Skip to content

Commit

Permalink
fix OpType check
Browse files Browse the repository at this point in the history
  • Loading branch information
zhanghonggeng committed Feb 13, 2025
1 parent 429b798 commit bb2423a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion paddle2onnx/mapper/tensor/elementwise.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}
Expand Down

0 comments on commit bb2423a

Please sign in to comment.