pytorch
39ff87c6 - [ROCM] Extend try-catch mechanism for ROCM detection (#99980)

Commit
1 year ago
[ROCM] Extend try-catch mechanism for ROCM detection (#99980) ROCM path detection currently relies on `hipconfig`. On some systems when calling `hipconfig` through `subprocess` python raises a `NotADirectoryError` that isn't catch at the moment. This commit adds `NotADirectoryError` to exceptions catched when calling `hipconfig`. Fixes #98629 Pull Request resolved: https://github.com/pytorch/pytorch/pull/99980 Approved by: https://github.com/jeffdaily, https://github.com/malfet
Author
Committer
Parents
Loading