.

pytorch · Oct 15, 2024 · de40d17 · de40d17
1 parent 82924d2
commit de40d17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchcodec/decoders/_core/VideoDecoderOps.cpp b/src/torchcodec/decoders/_core/VideoDecoderOps.cpp
@@ -168,7 +168,7 @@ void _add_video_stream(
   if (device.has_value()) {
     if (device.value() == "cpu") {
       options.device = torch::Device(torch::kCPU);
-    } else if (device.value().starts_with("cuda")) {
+    } else if (device.value().rfind("cuda", 0) == 0) { // starts with "cuda"
       std::string deviceStr(device.value());
       options.device = torch::Device(deviceStr);
     } else {