onnxruntime
e2abba18 - Skip softmax BF16 test for ROCm (#21162)

Commit
1 year ago
Skip softmax BF16 test for ROCm (#21162) ### Description Skip softmax BF16 test for ROCm, because BFloat16 is unsupported by MIOpen, and `torch.cuda.is_available()` also returns `True` for ROCm.
Author
Parents
Loading