From cf0a3471eed211e14a9afacf52dc5ca6fcf44668 Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Mon, 3 Nov 2025 10:35:00 -0800 Subject: [PATCH] Update test_persistent_kernels.expected stack-info: PR: https://github.com/pytorch/helion/pull/1070, branch: jansel/stack/214 --- test/test_persistent_kernels.expected | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_persistent_kernels.expected b/test/test_persistent_kernels.expected index efb27a4dd..8be817280 100644 --- a/test/test_persistent_kernels.expected +++ b/test/test_persistent_kernels.expected @@ -2092,7 +2092,7 @@ def test_kernel(x: torch.Tensor, *, _launcher=_default_launcher): _BLOCK_SIZE_1 = 16 # src[test_persistent_kernels.py:N]: for tile in hl.tile(x.size(), block_size=[32, 16]): # src[test_persistent_kernels.py:N]: result[tile] = x[tile] + 1 - _launcher(_helion_test_kernel, (_NUM_SM,), x, result, _NUM_SM, _BLOCK_SIZE_0, _BLOCK_SIZE_1, num_warps=4, num_stages=2) + _launcher(_helion_test_kernel, (_NUM_SM,), x, result, _NUM_SM, _BLOCK_SIZE_0, _BLOCK_SIZE_1, num_warps=4, num_stages=1) # src[test_persistent_kernels.py:N]: return result return result