Skip to content

Commit

Permalink
vsort/vs_onnxruntime.cpp: remove gridsample hack for cuda ep
Browse files Browse the repository at this point in the history
  • Loading branch information
WolframRhodium committed Apr 19, 2024
1 parent 9622eb4 commit 2a0b7bc
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions vsort/vs_onnxruntime.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ static std::mutex capture_lock;
// rename GridSample to com.microsoft::GridSample
// onnxruntime has support for CUDA-accelerated GridSample only in its own opset domain
static void rename(ONNX_NAMESPACE::ModelProto & model) {
#if ORT_API_VERSION < 18
constexpr auto ms_domain = "com.microsoft";

bool has_ms_opset = false;
Expand All @@ -95,6 +96,7 @@ static void rename(ONNX_NAMESPACE::ModelProto & model) {
*node.mutable_domain() = ms_domain;
}
}
#endif // ORT_API_VERSION < 18
}


Expand Down

0 comments on commit 2a0b7bc

Please sign in to comment.