fix: align github syntax for cuda (#1316)

janhq · Sep 24, 2024 · 5e7bda4 · 5e7bda4
1 parent ae8be75
commit 5e7bda4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/engine/services/engine_service.cc b/engine/services/engine_service.cc
@@ -119,9 +119,12 @@ void EngineService::UnzipEngine(const std::string& engine,
   CTL_INF("engine: " << engine);
   CTL_INF("CUDA version: " << hw_inf_.cuda_driver_version);
   std::string cuda_variant = "cuda-";
-  cuda_variant += GetSuitableCudaVersion(engine, hw_inf_.cuda_driver_version) +
-                  "-" + hw_inf_.sys_inf->os + "-" + hw_inf_.sys_inf->arch +
-                  ".tar.gz";
+  auto cuda_github =
+      GetSuitableCudaVersion(engine, hw_inf_.cuda_driver_version);
+  // Github release cuda example: cuda-12-0-windows-amd64.tar.gz
+  std::replace(cuda_github.begin(), cuda_github.end(), '.', '-');
+  cuda_variant += cuda_github + "-" + hw_inf_.sys_inf->os + "-" +
+                  hw_inf_.sys_inf->arch + ".tar.gz";
   CTL_INF("cuda_variant: " << cuda_variant);
 
   std::vector<std::string> variants;