Skip to content

Commit

Permalink
[AMD] Skip scaled_dot tests for gfx11 and gfx12 (#5008)
Browse files Browse the repository at this point in the history
`scaled_dot` is not yet implemented on `gfx11` and `gfx12`
so disable unit tests for now.
  • Loading branch information
AlexAUT authored Oct 29, 2024
1 parent 69f656c commit bf6bd0b
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions python/test/unit/language/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -3343,6 +3343,9 @@ def test_scaled_dot(M, N, K, col_a, col_b, type_a, type_b, num_warps, mma, kpack
pytest.skip(f"scaled_dot({type_a}, {type_b}) not yet implemented for HIP")
if mma == 16 and K == 64:
pytest.skip(f"K == {K} too small for mfma {mma} in scaled_dot")
arch = triton.runtime.driver.active.get_current_target().arch
if "gfx11" in arch or "gfx12" in arch:
pytest.skip("scaled_dot not yet implemented for gfx11 and gfx12")

@triton.jit
def dot_scale_kernel(a_base, stride_a0, stride_a1, a_scale, b_base, stride_b0, stride_b1, out,
Expand Down

0 comments on commit bf6bd0b

Please sign in to comment.