Update Cublas versions in Windows build and packaging scripts

locaal-ai · Mar 20, 2024 · 85914d2 · 85914d2
1 parent 6a2fcd2
commit 85914d2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.github/scripts/Build-Windows.ps1 b/.github/scripts/Build-Windows.ps1
@@ -4,7 +4,7 @@ param(
     [string] $Target = 'x64',
     [ValidateSet('Debug', 'RelWithDebInfo', 'Release', 'MinSizeRel')]
     [string] $Configuration = 'RelWithDebInfo',
-    [ValidateSet('cpu', 'cuda12_2', 'cuda11_8')]
+    [ValidateSet('cpu', '12.2.0', '11.8.0')]
     [string] $Cublas = 'cpu',
     [switch] $SkipAll,
     [switch] $SkipBuild,

diff --git a/.github/scripts/Package-Windows.ps1 b/.github/scripts/Package-Windows.ps1
@@ -4,7 +4,7 @@ param(
     [string] $Target = 'x64',
     [ValidateSet('Debug', 'RelWithDebInfo', 'Release', 'MinSizeRel')]
     [string] $Configuration = 'RelWithDebInfo',
-    [ValidateSet('cpu', 'cuda12_2', 'cuda11_8')]
+    [ValidateSet('cpu', '12.2.0', '11.8.0')]
     [string] $Cublas = 'cpu',
     [switch] $BuildInstaller,
     [switch] $SkipDeps
@@ -49,8 +49,9 @@ function Package {
     $BuildSpec = Get-Content -Path ${BuildSpecFile} -Raw | ConvertFrom-Json
     $ProductName = $BuildSpec.name
     $ProductVersion = $BuildSpec.version
+    $CudaName = "cuda${Cublas}"
 
-    $OutputName = "${ProductName}-${ProductVersion}-windows-${Target}-${Cublas}"
+    $OutputName = "${ProductName}-${ProductVersion}-windows-${Target}-${CudaName}"
 
     if ( ! $SkipDeps ) {
         Install-BuildDependencies -WingetFile "${ScriptHome}/.Wingetfile"

diff --git a/.github/workflows/build-project.yaml b/.github/workflows/build-project.yaml
@@ -227,7 +227,7 @@ jobs:
     needs: check-event
     strategy:
       matrix:
-        cublas: [cpu, cuda12_2, cuda11_8]
+        cublas: [cpu, 12.2.0, 11.8.0]
     defaults:
       run:
         shell: pwsh