Enable tests on py314 (#1306)

Author: oulgen

helion/_testing.py

@@ -199,11 +199,6 @@ def skipIfLowVRAM(
     return unittest.skipIf(low_vram, reason)
 
 
-def skipIfPy314(reason: str) -> Callable[[Callable], Callable]:
-    """Skip test if running on Python 3.14"""
-    return unittest.skipIf(sys.version_info >= (3, 14), reason)
-
-
 def skipIfPyTorchBaseVerLessThan(min_version: str) -> Callable[[Callable], Callable]:
     """Skip test if PyTorch base version is less than the specified version.
 

test/test_torch_compile.py

@@ -8,15 +8,13 @@
 from helion._testing import DEVICE
 from helion._testing import RefEagerTestBase
 from helion._testing import TestCase
-from helion._testing import skipIfPy314
 from helion._testing import skipIfRefEager
 from helion._testing import skipIfRocm
 import helion.language as hl
 
 
 class TestTorchCompile(RefEagerTestBase, TestCase):
     @skipIfRefEager("does not work with ref eager")
-    @skipIfPy314("torch.compile not yet supported on Python 3.14")
     @skipIfRocm("torch.compile add kernel missing kernel metadata fields on ROCm")
     def test_add_kernel(self):
         @helion.kernel(config=helion.Config(block_sizes=[1, 2]))

test/test_views.py

@@ -11,7 +11,6 @@
 from helion._testing import TestCase
 from helion._testing import code_and_output
 from helion._testing import skipIfCpu
-from helion._testing import skipIfPy314
 from helion._testing import skipIfRefEager
 from helion._testing import skipIfRocm
 import helion.language as hl
@@ -414,7 +413,6 @@ def foo(x: torch.Tensor) -> torch.Tensor:
         self.assertIn("tl.reshape", code)
         self.assertExpectedJournal(code)
 
-    @skipIfPy314("torch.compile not yet supported on Python 3.14")
     def test_stack_dim0(self):
         @helion.kernel(autotune_effort="none", static_shapes=True)
         def test_stack_dim0_kernel(