-
rasmith authored
[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
[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>