[minimax-m3] Split 1/4: sparse attention ops + JIT kernels + config foundation (#28712)

This commit is contained in:
Xinyuan Tong
2026-06-22 13:10:43 -07:00
committed by GitHub
parent b5e4e289b1
commit 7c23d2255a
51 changed files with 11157 additions and 33 deletions
@@ -248,8 +248,10 @@ class BenchmarkWorker:
torch.get_device_module().manual_seed_all(0)
self.seed = seed
# Get the device ID to allocate tensors and kernels
# on the respective GPU.
self.device_id = int(ray.get_gpu_ids()[0])
# on the respective GPU. Ray isolates each worker to a single visible
# GPU via CUDA_VISIBLE_DEVICES, so the local ordinal is always 0. On
# ROCm using the global ray gpu id here raises "invalid device ordinal".
self.device_id = 0 if is_hip() else int(ray.get_gpu_ids()[0])
set_global_server_args_for_scheduler(server_args)
def benchmark(