File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments