feat: add cuda core dump CI warpper (#18909)

This commit is contained in:
Liangsheng Yin
2026-02-17 14:49:26 -08:00
committed by GitHub
parent 9a7d6be567
commit 83a475e8d7
7 changed files with 272 additions and 2 deletions

View File

@@ -0,0 +1,29 @@
"""Smoke test: intentionally trigger a CUDA illegal memory access
to verify the coredump collection pipeline works end-to-end.
Manual use: python3 test/registered/debug_utils/test_cuda_coredump_smoke.py
"""
import unittest
import torch
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(
est_time=10,
suite="stage-a-test-1",
disabled="Manual only: triggers intentional CUDA crash for coredump verification",
)
class TestCudaCoredumpSmoke(unittest.TestCase):
def test_trigger_illegal_memory_access(self):
x = torch.zeros(10, device="cuda")
y = torch.arange(10, device="cuda")
x[y * y] = 1
torch.cuda.synchronize()
if __name__ == "__main__":
unittest.main()