github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

enable more gemm tests corresponding to PR#273 (#279)

Browse Source
This commit is contained in:
Shucai Xiao
2023-08-02 16:45:31 -05:00
committed by GitHub
parent a318bc5777
commit 31cfda8f0e
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/test/unit/language/test_core_amd.py
View File

@@ -1280,9 +1280,6 @@ def test_dot(M, N, K, num_warps, col_a, col_b, epilogue, allow_tf32, dtype, devi
if capability[0] == 7:
if (M, N, K, num_warps) == (128, 256, 32, 8):
pytest.skip("shared memory out of resource")
if torch.version.hip is not None:
if (M, N, K) == (64, 128, 128):
pytest.skip("Not supported: memory out of resource.")
torch.backends.cuda.matmul.allow_tf32 = allow_tf32

2
python/test/unit/language/test_gemm.py
View File

@@ -112,6 +112,8 @@ def get_variant_golden(a, b):
[32, 128, 256, 4, 32, 128, 64],
[64, 128, 64, 4, 64, 128, 32],
[64, 64, 128, 4, 64, 64, 32],
[128, 128, 64, 4, 128, 128, 32],
[128, 128, 128, 4, 128, 128, 32],
[128, 64, 64, 4, 128, 64, 32],
[128, 64, 128, 4, 128, 64, 32],
[64, 64, 256, 4, 64, 64, 64],