[CI] Add entrypoint to hc_combine test for standalone execution (#37406)

This commit is contained in:
Even Zhou
2026-09-01 22:08:00 +08:00
committed by GitHub
parent 3ae54c6ca2
commit 00689c0c94
@@ -1,3 +1,5 @@
import sys
import pytest
import torch
@@ -51,3 +53,7 @@ def test_hc_combine_strided_pre():
ref = _reference(x_flat, pre, hc, torch.bfloat16)
scale = ref.float().abs().max().clamp(min=1e-6)
assert (got.float() - ref.float()).abs().max() / scale < 5e-2
if __name__ == "__main__":
sys.exit(pytest.main([__file__]))