[Bugfix][ROCm] Using device_type because on ROCm the API is still torch.cuda (#17601)
Signed-off-by:
Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Showing
Please register or sign in to comment
Signed-off-by:
Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>