Skip to content

Commit b7a0502

Browse files
[AMD] Fix test_split_subview on gfx11/gfx12 (#7513)
This was fixed by #7457 but then partially overwritten by #7480. Co-authored-by: Paul Trojahn <[email protected]>
1 parent 8e52b2e commit b7a0502

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/test/unit/language/test_core.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6360,8 +6360,7 @@ def test_local_load_store(M, N, K, dist_layout, shared_layout, device, tmp_path:
63606360
def test_split_subview(M, N, M_tile_size, N_tile_size, device='cuda'):
63616361
if not is_hip() and N_tile_size == 32:
63626362
pytest.skip("Will fix spliting along the swizzling pattern in the next PR")
6363-
threads_per_warp = 64 if is_hip() else 32
6364-
num_rows_per_warp = threads_per_warp // 4
6363+
num_rows_per_warp = THREADS_PER_WARP // 4
63656364
num_repeats_M = triton.cdiv(M, M_tile_size)
63666365
num_repeats_N = triton.cdiv(N, N_tile_size)
63676366

0 commit comments

Comments
 (0)