We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 72f1fe4 commit 88e9ae6Copy full SHA for 88e9ae6
modules/launch_utils.py
@@ -534,8 +534,8 @@ def prepare_environment():
534
if args.use_rocm or args.use_zluda:
535
print('No ROCm agent was found. Please make sure that graphics driver is installed and up to date.')
536
backend = "cpu"
537
+ args.skip_torch_cuda_test = True
538
else:
- print('ROCm: AMD toolkit detected')
539
print(f'ROCm: agents={[gpu.name for gpu in amd_gpus]}')
540
if args.device_id is None:
541
index = 0
0 commit comments