debug import torch timing

Author: ryan-williams

.github/workflows/test.yaml

@@ -46,7 +46,10 @@ jobs:
       - name: Check GPU and PyTorch
         run: |
           nvidia-smi
-          python -c "import torch; print(f'PyTorch: {torch.__version__}, CUDA: {torch.cuda.is_available()}, version: {torch.version.cuda}')"
+          echo "Timing PyTorch import..."
+          time python -c "import torch; print(f'PyTorch: {torch.__version__}, CUDA: {torch.cuda.is_available()}, version: {torch.version.cuda}')"
+          echo "Second import (should be cached):"
+          time python -c "import torch; print('Imported torch')"
       - name: Install mamba-ssm and test dependencies
         run: |
           # Use all available CPUs for compilation (we're only building for 1 GPU arch)