pytorch
a305c078 - [JIT] Prevent nvfuser registration on ROCm

Commit
2 years ago
[JIT] Prevent nvfuser registration on ROCm Previously, cuda_graph_fuser.h registration of the nvfuser pass used `at::globalContext().hasHIP()` to check whether we were using ROCm/HIP. However, I don't think that check actually does anything; on the ROCm CI jobs the registration would still succeed. Instead it's replaced with `#ifdef USE_ROCM`. Verified this by enabling the NVFuser tests on ROCm and running in CI. Before this change: the NVFuser test in CI on ROCm would throw really long and complex errors. Now, it errors out immediately when trying to enable nvfuser. Pull Request resolved: https://github.com/pytorch/pytorch/pull/75284 Approved by: https://github.com/eellison
Author
Committer
Parents
Loading