benchmarks/kernels/benchmark_moe.py · 5dd8df070172ac20e99a7dbd3d96cb6b054f0f57 · OpenDAS / vllm_cscc · GitLab

"docs/backends/vllm/vllm-omni.md" did not exist on "21abf0768664e91f8351092dc9990fe9a2f93e27"

Find file Blame History Permalink

replace `with torch.cuda.device` with `with torch.accelerator.device_index` (#36144) · 894843eb
Yan Ma authored Mar 12, 2026
```
Signed-off-by: Yan Ma <yan.ma@intel.com>
```
894843eb

benchmark_moe.py 35.3 KB