tests/lora/test_punica_ops.py · b188bab4417f7f94df0c4e84399d163e6c0db316 · OpenDAS / vllm_cscc

[CI][AMD][BugFix] Add torch.cuda.set_device to test_punica_ops so punica... · b188bab4

rasmith authored Feb 25, 2026


[CI][AMD][BugFix] Add  torch.cuda.set_device to test_punica_ops so punica kernels execute on same device as tensor (#34985)
Signed-off-by: Randall Smith <Randall.Smith@amd.com>

b188bab4

test_punica_ops.py 11.1 KB

Replace test_punica_ops.py