From 9c67ca8249e2fb163c44689886c558e49bba5153 Mon Sep 17 00:00:00 2001 From: dou <15529241576@163.com> Date: Tue, 24 Dec 2024 02:23:34 +0000 Subject: [PATCH] delete the supported domain version upper bounds --- onnxruntime/core/providers/cann/cann_graph.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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;