Skip to content

Commit b0288ed

Browse files
authored
[ViPPET] Fix GPU device name check to support multiple GPU identifiers (#1501)
1 parent 72475e5 commit b0288ed

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)