Skip to content

Commit

Permalink
delete the supported domain version upper bounds
Browse files Browse the repository at this point in the history
  • Loading branch information
bachelor-dou committed Dec 24, 2024
1 parent 81cd6ea commit 9c67ca8
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions onnxruntime/core/providers/cann/cann_graph.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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;

Expand Down Expand Up @@ -62,7 +61,7 @@ std::vector<NodeIndex> 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;
Expand Down

0 comments on commit 9c67ca8

Please sign in to comment.