Tests: Fix device list of benchmark script only showing a single GPU
Pull Request #104583
This commit is contained in:
parent
7351f533e0
commit
8a32d56056
|
@ -39,7 +39,6 @@ def get_gpu_device(args: None) -> List:
|
|||
if device.type == device_type:
|
||||
result.append({'type': device.type, 'name': device.name, 'index': index})
|
||||
index += 1
|
||||
break
|
||||
|
||||
return result
|
||||
|
||||
|
|
Loading…
Reference in New Issue