pinokiod 5.1.30 → 5.1.32

package/kernel/bin/cuda.js

@@ -1,7 +1,33 @@
 const semver = require('semver')
 const fs = require('fs')
 const path = require('path')
+const { glob } = require('glob')
 class Cuda {
+  async hasNvTargetHeader() {
+    const prefix = this.kernel.bin.path("miniconda")
+    const patterns = [
+      "Library/include/nv/target",
+      "include/nv/target",
+      "Library/targets/*/include/nv/target",
+      "targets/*/include/nv/target",
+    ]
+    for (const pattern of patterns) {
+      if (pattern.includes("*")) {
+        try {
+          const matches = await glob(pattern, { cwd: prefix, absolute: true, nodir: true })
+          if (matches && matches.length > 0) {
+            return true
+          }
+        } catch (e) {}
+      } else {
+        try {
+          await fs.promises.stat(path.resolve(prefix, pattern))
+          return true
+        } catch (e) {}
+      }
+    }
+    return false
+  }
   async patchCudaActivationScript() {
     if (this.kernel.platform !== "win32") {
       return
@@ -60,7 +86,7 @@ class Cuda {
         await this.kernel.bin.exec({
           message: [
             "conda clean -y --all",
-            "conda install -y nvidia/label/cuda-12.8.1::cuda nvidia/label/cuda-12.8.1::cuda-cccl"
+            "conda install -y nvidia/label/cuda-12.8.1::cuda conda-forge::cuda-cccl"
           ]
         }, ondata)
         await this.stashActivationScripts()
@@ -98,7 +124,7 @@ class Cuda {
             console.log("cuda version", coerced)
             if (semver.satisfies(coerced, ">=12.8.1")) {
               console.log("cuda satisfied")
-              if (!this.kernel.bin.installed.conda.has("cuda-cccl")) {
+              if (!(await this.hasNvTargetHeader())) {
                 return false
               }
               let deactivate_list = [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pinokiod",
-  "version": "5.1.30",
+  "version": "5.1.32",
   "description": "",
   "main": "index.js",
   "scripts": {