Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
chilo-ms committed Oct 23, 2023
1 parent a66d399 commit 0d59309
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions onnxruntime/core/session/inference_session.cc
Original file line number Diff line number Diff line change
Expand Up @@ -627,16 +627,16 @@ common::Status InferenceSession::RegisterExecutionProvider(const std::shared_ptr
for (auto candidate_custom_op_domain : candidate_custom_op_domains) {
for (auto registry_kernel : registry_kernels) {
const auto& kernel_map = registry_kernel->GetKernelCreateMap();
bool need_resigter = true;
bool need_register = true;
// If the kernel registry is the ep's custom op registry, we only need to check the first kernel,
// because all kernels in one kernel registry should have the same domain name.
for (auto iter = kernel_map.begin(); iter != kernel_map.end(); iter++) {
if (iter->second.kernel_def->Domain() == candidate_custom_op_domain->domain_) {
need_resigter = false;
need_register = false;
break;
}
}
if (need_resigter) {
if (need_register) {
custom_op_domains.push_back(candidate_custom_op_domain);
}
}
Expand Down

0 comments on commit 0d59309

Please sign in to comment.