fix rocm hip mismatch to cuda
All checks were successful
Build and Publish Docker Images / build-cuda (push) Successful in 4m52s
Build and Publish Docker Images / build-rocm (push) Successful in 6m59s
Build and Publish Docker Images / build-cpu (push) Successful in 24m31s

This commit is contained in:
Stephen Tafoya
2026-06-13 02:33:57 -06:00
parent 4ad64ad1a8
commit a0ff64629c
2 changed files with 3 additions and 2 deletions

View File

@@ -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)