tests/kernels/attention/test_flashmla.py · 53ec16a705f27dc72d5b824a5b7ccd490f235383 · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145) · 53ec16a7
Kunshang Ji authored Mar 12, 2026
```
Signed-off-by: Kunshang Ji <jikunshang95@gmail.com>
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
```
53ec16a7

test_flashmla.py 6.02 KB