minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

Fix CI caused by https://github.com/sgl-project/sglang/pull/29576 (#29596)

Browse Source 
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
This commit is contained in:
Brayden Zhong
2026-06-28 23:46:18 -07:00
committed by GitHub
co-authored by Brayden Zhong
parent c0d45cda29
commit c7b9b92d9a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/test_dsa_indexer.py
+1 -1
View File
@@ -321,8 +321,8 @@ class TestDSAIndexer(CustomTestCase):
params.update(kwargs) params.update(kwargs)
torch.set_default_dtype(self.dtype) torch.set_default_dtype(self.dtype)
with torch.device(self.device):
indexer = Indexer(**params) indexer = Indexer(**params)
# Move indexer to CUDA device
indexer = indexer.to(device=self.device) indexer = indexer.to(device=self.device)
# Convert linear layer weights to bfloat16 (but preserve LayerNorm's float32 # Convert linear layer weights to bfloat16 (but preserve LayerNorm's float32