Fix actions test loading library

Author: 2kai2kai2

.github/workflows/npm-publish.yml

@@ -24,13 +24,20 @@ jobs:
     runs-on: ${{ matrix.image }}
     steps:
       - uses: actions/checkout@v3
-      - uses: actions/setup-node@v2
+      - uses: actions/setup-node@v4
         with:
           node-version: '16'
       - run: |
           sudo apt-get install -y nvidia-driver-535
           sudo apt-get install -y libnvidia-ml-dev
         if: matrix.kernel == 'linux-x64'
+      - uses: conda-incubator/setup-miniconda@v3
+        with:
+          activate-environment: nvml-tools
+        if: matrix.kernel == 'win32-x64'
+      - run: |
+          conda install cuda-nvml-dev -c nvidia/label/cuda-12.9.1
+        if: matrix.kernel == 'win32-x64'
       - run: npm run install-dependencies
       - run: npm run prebuild
       - uses: actions/upload-artifact@v4
@@ -76,19 +83,22 @@ jobs:
     runs-on: ${{ matrix.image }}
     needs: [pack-module]
     steps:
-      - uses: actions/setup-node@v2
       - uses: actions/checkout@v3
         name: Fetch tests
         with:
           sparse-checkout: |
             test/
           sparse-checkout-cone-mode: false
+      - uses: actions/setup-node@v2
       - uses: actions/download-artifact@v4
         name: Load bundled module
         id: download-module
         with:
           name: package
-      - run: npm i ${{ needs.pack-module.outputs.tgz_name }}
+      - run: ls
+      - run: |
+          npm init -y
+          npm i ${{ needs.pack-module.outputs.tgz_name }}
       
       - name: Basic CPU Test | Build Test
         run: npm exec tsc ./test/test_cpu_basic.ts

binding.gyp

@@ -10,15 +10,27 @@
                     ],
                     "include_dirs": [
                         "./node_modules/node-addon-api",
-                        "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.9/include"
+                        "<!(echo %CUDA_PATH%)/include",
+                        "<!(echo %CONDA_PREFIX%)/Library/include",
                     ],
                     "ldflags": [
                         "-lpsapi", # gcc/g++
                         "-lnvidia-ml",
                     ],
                     "link_settings": {
                         "libraries": [
-                            "psapi.lib" # msvc
+                            "psapi.lib", # msvc
+                        ],
+                        "variables": {
+                            'has_cuda_path': '<!(if defined CUDA_PATH (echo 1) else (echo 0))',
+                        },
+                        "conditions": [
+                            ['has_cuda_path==1', { # normal cuda toolkit install (used locally)
+                                "libraries": ["<!(echo %CUDA_PATH%)/lib/x64/nvml.lib"],
+                            }],
+                            ['has_cuda_path==0', { # conda (used in github actions)
+                                "libraries": ["<!(echo %CONDA_PREFIX%)/Library/lib/nvml.lib"],
+                            }],
                         ]
                     },
                     "cflags!": [ "-fno-exceptions" ],

package-lock.json

@@ -5,7 +5,7 @@
     "name": "kaih2o",
     "url": "https://github.com/2kai2kai2"
   },
-  "version": "1.4.1",
+  "version": "1.5.0",
   "keywords": [
     "process",
     "processes",
@@ -25,7 +25,7 @@
   },
   "dependencies": {
     "node-gyp": "^10.0.1",
-    "node-gyp-build": "^4.8.4"
+    "node-gyp-build": "^4.2.0"
   },
   "devDependencies": {
     "@types/node": "^16.9.6",
@@ -46,4 +46,4 @@
   },
   "type": "commonjs",
   "types": "index.d.ts"
-}
+}