[CI] Move existing unit tests into unit directory (#20631)

test/registered/unit/layers/test_conv_layer.py

@@ -1,3 +1,7 @@
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=10, suite="stage-b-test-small-1-gpu")
+
 import unittest
 
 import torch

test/registered/unit/layers/test_mamba_state_scatter_triton.py

@@ -1,3 +1,7 @@
+from sglang.test.ci.ci_register import register_cuda_ci
+
+register_cuda_ci(est_time=10, suite="stage-b-test-small-1-gpu")
+
 import os
 import unittest
 
@@ -177,10 +181,10 @@ class TestMambaStateScatterCorrectness(unittest.TestCase):
         ssm_elems = 1024
         conv_elems = 512
 
-        ssm_states0 = torch.empty(
+        ssm_states0 = torch.randn(
             (L, C, ssm_elems), device=device, dtype=torch.bfloat16
         )
-        conv_states0 = torch.empty(
+        conv_states0 = torch.randn(
             (L, C, conv_elems), device=device, dtype=torch.bfloat16
         )
         intermediate_ssm = torch.randn(

test/registered/unit/mem_cache/test_radix_cache_slru_accuracy.py

@@ -20,7 +20,8 @@ class TestSLRUAccuracy(unittest.TestCase):
 
     def setUp(self):
         """Setup minimal memory pools for testing"""
-        device = "cpu"  # Using CPU for testing simplicity
+        torch.set_default_device(None)
+        device = "cpu"
         dtype = torch.float16
 
         # Create smaller KV cache to ensure evictions occur

test/registered/unit/utils/test_gauge_histogram.py

@@ -1,3 +1,7 @@
+from sglang.test.ci.ci_register import register_cpu_ci
+
+register_cpu_ci(est_time=5, suite="stage-a-cpu-only")
+
 import unittest
 
 from sglang.srt.utils.gauge_histogram import BucketLabels