pytorch
0a4e4de5 - [ROCm] add case for FP32MatMulPattern skip property (#84077)

Commit
2 years ago
[ROCm] add case for FP32MatMulPattern skip property (#84077) TF32 is not supported on ROCm and hence the torch/profiler/_pattern_matcher.py FP32MatMulPattern should return False for ROCm instead of checking the results of torch.cuda.get_arch_list(). Depending on the gfx arch running the test, test_profiler.py's test_profiler_fp32_matmul_pattern (__main__.TestExperimentalUtils) will fail otherwise. Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/84077 Approved by: https://github.com/jeffdaily, https://github.com/kit1980
Author
Committer
Parents
Loading