Add cuBLAS support to build and package actions

Author: royshil

.github/actions/build-plugin/action.yaml

@@ -8,6 +8,10 @@ inputs:
     description: 'Build configuration'
     required: false
     default: 'RelWithDebInfo'
+  cublas:
+    description: 'Enable cuBLAS'
+    required: false
+    default: 'cpu'
   codesign:
     description: 'Enable codesigning (macOS only)'
     required: false
@@ -78,6 +82,7 @@ runs:
         $BuildArgs = @{
           Target = '${{ inputs.target }}'
           Configuration = '${{ inputs.config }}'
+          Cublas = '${{ inputs.cublas }}'
         }
 
         .github/scripts/Build-Windows.ps1 @BuildArgs

.github/actions/package-plugin/action.yaml

@@ -8,6 +8,10 @@ inputs:
     description: 'Build configuration'
     required: false
     default: 'RelWithDebInfo'
+  cublas:
+    description: 'Enable cuBLAS'
+    required: false
+    default: 'cpu'
   codesign:
     description: 'Enable codesigning (macOS only)'
     required: false
@@ -108,6 +112,7 @@ runs:
         $PackageArgs = @{
           Target = '${{ inputs.target }}'
           Configuration = '${{ inputs.config }}'
+          Cublas = '${{ inputs.cublas }}'
         }
 
         if ( '${{ inputs.package }}' -eq 'true' ) {

.github/scripts/Build-Windows.ps1

@@ -4,6 +4,8 @@ param(
     [string] $Target = 'x64',
     [ValidateSet('Debug', 'RelWithDebInfo', 'Release', 'MinSizeRel')]
     [string] $Configuration = 'RelWithDebInfo',
+    [ValidateSet('cpu', 'cuda12_2', 'cuda11_8')]
+    [string] $Cublas = 'cpu',
     [switch] $SkipAll,
     [switch] $SkipBuild,
     [switch] $SkipDeps,
@@ -77,6 +79,13 @@ function Build {
             '--preset', $Preset
         )
 
+        if ( $Cublas -ne 'cpu' ) {
+            $CmakeArgs += @(
+                '-DLOCALVOCAL_WITH_CUDA=ON',
+                "-DCUDA_TOOLKIT_ROOT_DIR=$Env:CUDA_TOOLKIT_ROOT_DIR"
+            )
+        }
+
         $CmakeBuildArgs += @(
             '--build'
             '--preset', $Preset

.github/scripts/Package-Windows.ps1

@@ -4,6 +4,8 @@ param(
     [string] $Target = 'x64',
     [ValidateSet('Debug', 'RelWithDebInfo', 'Release', 'MinSizeRel')]
     [string] $Configuration = 'RelWithDebInfo',
+    [ValidateSet('cpu', 'cuda12_2', 'cuda11_8')]
+    [string] $Cublas = 'cpu',
     [switch] $BuildInstaller,
     [switch] $SkipDeps
 )
@@ -48,7 +50,7 @@ function Package {
     $ProductName = $BuildSpec.name
     $ProductVersion = $BuildSpec.version
 
-    $OutputName = "${ProductName}-${ProductVersion}-windows-${Target}"
+    $OutputName = "${ProductName}-${ProductVersion}-windows-${Target}-${Cublas}"
 
     if ( ! $SkipDeps ) {
         Install-BuildDependencies -WingetFile "${ScriptHome}/.Wingetfile"

.github/workflows/build-project.yaml

@@ -225,6 +225,9 @@ jobs:
     name: Build for Windows 🪟
     runs-on: windows-2022
     needs: check-event
+    strategy:
+      matrix:
+        cublas: [cpu, cuda12_2, cuda11_8]
     defaults:
       run:
         shell: pwsh
@@ -234,6 +237,18 @@ jobs:
           submodules: recursive
           fetch-depth: 0
 
+      - name: Install CUDA Toolkit
+        if: ${{ matrix.cublas != 'cpu' }}
+        id: cuda-toolkit
+        uses: Jimver/cuda-toolkit@v0.2.11
+        with:
+          cuda: '${{ matrix.cublas }}'
+
+      - name: Set CUDA_TOOLKIT_ROOT_DIR if CUDA is installed
+        if: ${{ matrix.cublas != 'cpu' }}
+        run: |
+          "CUDA_TOOLKIT_ROOT_DIR=${{ steps.cuda-toolkit.outputs.cudaPath }}" >> $env:GITHUB_ENV
+
       - name: Set Up Environment 🔧
         id: setup
         run: |
@@ -254,16 +269,18 @@ jobs:
         with:
           target: x64
           config: ${{ needs.check-event.outputs.config }}
+          cublas: ${{ matrix.cublas }}
 
       - name: Package Plugin 📀
         uses: ./.github/actions/package-plugin
         with:
           target: x64
           config: ${{ needs.check-event.outputs.config }}
           package: ${{ fromJSON(needs.check-event.outputs.package) }}
+          cublas: ${{ matrix.cublas }}
 
       - name: Upload Artifacts 📡
         uses: actions/upload-artifact@v3
         with:
-          name: ${{ steps.setup.outputs.pluginName }}-${{ steps.setup.outputs.pluginVersion }}-windows-x64-${{ needs.check-event.outputs.commitHash }}
+          name: ${{ steps.setup.outputs.pluginName }}-${{ steps.setup.outputs.pluginVersion }}-windows-x64-${{ matrix.cublas }}-${{ needs.check-event.outputs.commitHash }}
           path: ${{ github.workspace }}/release/${{ steps.setup.outputs.pluginName }}-${{ steps.setup.outputs.pluginVersion }}-windows-x64*.*