mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-20 19:12:24 +00:00
6f566562ce
### Description <!-- Describe your changes. --> According to the pr #19229 supporting cuda EP use external compute stream, we add support for rocm EP. And when we testing this feature with torch, we found torch use stream 0 for the default stream, and `torch.cuda.current_stream()` returns `0` for current stream, but ort treat `0` or `nullptr` as invalid, and reset has_user_compute_stream to false. Will remove has_user_compute_stream option in the future. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> The motivation for this pr is that we want to use torch.cuda.graph to capture ort running kernel, which requires torch and ort are running in the same stream, so we use this API to set ort's working stream. |
||
|---|---|---|
| .. | ||
| codegen | ||
| common | ||
| dll | ||
| dlpack | ||
| eager | ||
| flatbuffers | ||
| framework | ||
| graph | ||
| language_interop_ops | ||
| mickey | ||
| mlas | ||
| optimizer | ||
| platform | ||
| providers | ||
| quantization | ||
| session | ||
| util | ||