From c7b9b92d9a57baa75339d54b882aac8a8bb2227d Mon Sep 17 00:00:00 2001 From: Brayden Zhong Date: Sun, 28 Jun 2026 23:46:18 -0700 Subject: [PATCH] Fix CI caused by https://github.com/sgl-project/sglang/pull/29576 (#29596) Co-authored-by: Brayden Zhong --- test/registered/kernels/test_dsa_indexer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/registered/kernels/test_dsa_indexer.py b/test/registered/kernels/test_dsa_indexer.py index c5cc81e47..54c4a2108 100644 --- a/test/registered/kernels/test_dsa_indexer.py +++ b/test/registered/kernels/test_dsa_indexer.py @@ -321,8 +321,8 @@ class TestDSAIndexer(CustomTestCase): params.update(kwargs) torch.set_default_dtype(self.dtype) - indexer = Indexer(**params) - # Move indexer to CUDA device + with torch.device(self.device): + indexer = Indexer(**params) indexer = indexer.to(device=self.device) # Convert linear layer weights to bfloat16 (but preserve LayerNorm's float32