Skip to content

Commit c3a59f2

Browse files
authored
skip gather_gemv code check for B200 and fb_code (#666)
1 parent b15b026 commit c3a59f2

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

test/test_examples.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
from helion._testing import TestCase
1313
from helion._testing import check_example
1414
from helion._testing import import_path
15-
from helion._testing import is_cuda
1615
from helion._testing import skipIfRefEager
1716
from helion._testing import skipIfRocm
1817

@@ -1195,7 +1194,7 @@ def test_gather_gemv(self):
11951194
def expected(w, idx, x):
11961195
return w[idx].to(x.dtype) @ x
11971196

1198-
code = check_example(
1197+
check_example(
11991198
"gather_gemv",
12001199
args,
12011200
expected(*args),
@@ -1205,9 +1204,6 @@ def expected(w, idx, x):
12051204
num_stages=1,
12061205
)
12071206

1208-
if is_cuda():
1209-
self.assertExpectedJournal(code)
1210-
12111207
def test_int4_gemm(self):
12121208
# Matrix dimensions
12131209
M, K, N = 256, 512, 256

0 commit comments

Comments
 (0)