pytorch
98012e4a - [ROCm] hipGraph support for pytorch mainline (#88202)

Commit
2 years ago
[ROCm] hipGraph support for pytorch mainline (#88202) With the release of ROCm 5.3 hip now supports a hipGraph implementation. All necessary backend work and hipification is done to support the same functionality as cudaGraph. Unit tests are modified to support a new TEST_GRAPH feature which allows us to create a single check for graph support instead of attempted to gather the CUDA level in annotations for every graph test Pull Request resolved: https://github.com/pytorch/pytorch/pull/88202 Approved by: https://github.com/jithunnair-amd, https://github.com/pruthvistony, https://github.com/malfet
Author
Committer
Parents
Loading