[kernel slimming] Move fast_hadamard_transform to jit_kernel (#18475)

2026-02-14 23:06:21 +08:00
parent ae95869292
commit c29394e3c8
17 changed files with 2170 additions and 71 deletions
@@ -584,8 +584,8 @@ class TestNSAIndexer(CustomTestCase):
            output = rotate_activation(x)
            self.assertEqual(output.shape, x.shape)
            self.assertEqual(output.dtype, torch.bfloat16)
-        except ImportError:
-            self.skipTest("sgl_kernel not available for hadamard_transform")
+        except Exception:
+            self.skipTest("hadamard JIT kernel not available")

    def test_rotate_activation_invalid_size(self):
        """Test that rotate_activation fails with non-power-of-2 size."""