diff --git a/onnxruntime/core/providers/cann/cann_graph.cc b/onnxruntime/core/providers/cann/cann_graph.cc index b8584d022ee51..23186a77dd65e 100644 --- a/onnxruntime/core/providers/cann/cann_graph.cc +++ b/onnxruntime/core/providers/cann/cann_graph.cc @@ -11,7 +11,6 @@ namespace onnxruntime { namespace cann { static int lower_bound = 8; // Supported domain version lower bounds -static int upper_bound = 15; // Supported domain version upper bounds std::once_flag flag; @@ -62,7 +61,7 @@ std::vector SupportONNXModel(const GraphViewer& graph_viewer) { for (const auto& index : graph_viewer.GetNodesInTopologicalOrder()) { const auto& node = graph_viewer.GetNode(index); - if (node->Domain() != kOnnxDomain || domain_version < lower_bound || domain_version > upper_bound || + if (node->Domain() != kOnnxDomain || domain_version < lower_bound || !cann_supported_ops.count(node->OpType())) { unsupported_nodes.push_back(index); continue;