[Tiny]Fix 1-gpu nightly test bugs (#13389)

2025-11-16 15:54:17 -08:00
parent 3ccd7fa669
commit d64dd3e18e
3 changed files with 4 additions and 19 deletions
@@ -239,6 +239,7 @@ class TestNSAIndexer(CustomTestCase):
        }
        params.update(kwargs)

+        torch.set_default_dtype(self.dtype)
        indexer = Indexer(**params)
        # Move indexer to CUDA device
        indexer = indexer.to(device=self.device)