fix rocm hip mismatch to cuda
This commit is contained in:
@@ -31,7 +31,7 @@ async def main() -> None:
|
||||
args = parser.parse_args()
|
||||
|
||||
device_map = {
|
||||
"rocm": "hip",
|
||||
"rocm": "cuda",
|
||||
}
|
||||
device = device_map.get(args.device, args.device)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user