Skip to content

Commit dc3281a

Browse files
committed
Fix GPU device name check to support multiple GPU identifiers
1 parent 693eea2 commit dc3281a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/visual-pipeline-and-platform-evaluation-tool/vippet/video_encoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def select_element(
153153
Selected encoder element string with properties, or None if not found
154154
"""
155155
key = OTHER
156-
if encoder_device.device_name == "GPU":
156+
if encoder_device.device_name.startswith("GPU"):
157157
if encoder_device.gpu_id == 0:
158158
key = GPU_0
159159
elif encoder_device.gpu_id is not None and encoder_device.gpu_id > 0:

0 commit comments

Comments
 (0)