Skip to content

Commit 57a1a9b

Browse files
committed
[intel] update 'make_default_opt_flags_intel' signature and xfail 'test_small_batch_matmul' after '9801a7a'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 5cb975c commit 57a1a9b

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

python/triton_kernels/tests/test_matmul.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
# testing utilities
2222
from triton_kernels.testing import assert_close, compute_actual_scale
2323
# target-specific utilities
24-
from triton_kernels.target_info import is_hip, is_hip_cdna3, is_cuda, is_hip_cdna4
24+
from triton_kernels.target_info import is_hip, is_xpu, is_hip_cdna3, is_cuda, is_hip_cdna4
2525

2626
# ---------------
2727
# initialize data
@@ -479,6 +479,8 @@ def round_x(x, idx):
479479
def test_small_batch_matmul(m, n, k):
480480
if is_hip():
481481
pytest.skip("Not fully tested on AMD")
482+
if is_xpu():
483+
pytest.xfail("Enable: https://github.com/intel/intel-xpu-backend-for-triton/issues/5092")
482484

483485
if m * n * k > 16384:
484486
pytest.skip()

python/triton_kernels/triton_kernels/matmul_ogs_details/opt_flags.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def make_default_opt_flags_intel(
3535
lhs_dtype,
3636
rhs_dtype,
3737
precision_config,
38+
batch_size,
3839
m,
3940
n,
4041
k,

0 commit comments

Comments
 (0)