Skip to content

Commit 2d7c3b5

Browse files
committed
Replace use_default_config in new test cases.
1 parent c3286f2 commit 2d7c3b5

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/test_errors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def torch_nonzero_in_device_code(x: torch.Tensor) -> torch.Tensor:
296296
def test_torch_chunk_device_error(self):
297297
"""Test that torch.chunk raises error in device loops and suggests hl.split()."""
298298

299-
@helion.kernel(use_default_config=True, static_shapes=True)
299+
@helion.kernel(autotune_effort="none", static_shapes=True)
300300
def kernel_with_chunk(q: torch.Tensor) -> torch.Tensor:
301301
_, _, M, D = q.shape
302302
D = hl.specialize(D)
@@ -330,7 +330,7 @@ def kernel_with_chunk(q: torch.Tensor) -> torch.Tensor:
330330
def test_torch_unbind_device_error(self):
331331
"""Test that torch.unbind raises error in device loops and suggests hl.split()."""
332332

333-
@helion.kernel(use_default_config=True, static_shapes=True)
333+
@helion.kernel(autotune_effort="none", static_shapes=True)
334334
def kernel_with_unbind(q: torch.Tensor) -> torch.Tensor:
335335
_, _, M, D = q.shape
336336
D = hl.specialize(D)

test/test_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def test_matmul_packed_int4_block_size_constexpr(self):
203203
torch.manual_seed(0)
204204
M = N = K = 32
205205

206-
@helion.kernel(use_default_config=True, static_shapes=True)
206+
@helion.kernel(autotune_effort="none", static_shapes=True)
207207
def matmul_bf16_packed_int4(
208208
A: torch.Tensor, B_packed: torch.Tensor, C: torch.Tensor
209209
) -> torch.Tensor:

0 commit comments

Comments
 (0)