pinokiod 7.5.11 → 7.5.13

package/kernel/index.js

@@ -95,6 +95,7 @@ class Kernel {
     this.torch_backend = "cpu"
     this.vram = 0
     this.ram = 0
+    this.gpu_driver = null
     this.readyState = new ReadyState(this)
     this.app_ready_status = this.readyState.status
     this.launchRequirements = new LaunchRequirements(this)
@@ -1347,6 +1348,7 @@ class Kernel {
         this.sysinfo = info
         this.gpu = info.gpu
         this.gpu_model = info.gpu_model
+        this.gpu_driver = info.gpu_driver || null
         this.torch_backend = info.torch_backend || "cpu"
         this.gpus = info.gpus
         this.vram = typeof info.vram === "number" ? info.vram : 0

package/kernel/sysinfo.js

@@ -49,7 +49,8 @@ class Sysinfo {
       gpus = controllers.map((x) => {
         return {
           name: (x.vendor || "").toLowerCase(),
-          model: (x.model || "").toLowerCase()
+          model: (x.model || "").toLowerCase(),
+          driver: x.driverVersion ? String(x.driverVersion) : null
         }
       })
     } else {
@@ -66,6 +67,9 @@ class Sysinfo {
     let model = (controller) => {
       return (controller && controller.model ? controller.model : "").toLowerCase()
     }
+    let driver = (controller) => {
+      return controller && controller.driverVersion ? String(controller.driverVersion) : null
+    }
 
     let is_nvidia = controllers.find(x => /nvidia/i.test(x.vendor || ""))
     let is_amd = bestByVram(controllers.filter(x => /(amd|advanced micro devices)/i.test(x.vendor || "")))
@@ -103,12 +107,14 @@ class Sysinfo {
 
     let vramMB = primaryController && primaryController.vram ? primaryController.vram : 0
     let vramGB = vramMB > 0 ? Math.round(vramMB / 1024) : 0
+    let gpu_driver = driver(primaryController)
     let torch_backend = await this.torch_backend({ is_nvidia, is_amd, is_apple, is_intel, gpu_model })
 
     this.info.graphics = g
     this.info.gpus = gpus
     this.info.gpu = gpu
     this.info.gpu_model = gpu_model
+    this.info.gpu_driver = gpu_driver
     this.info.torch_backend = torch_backend
     this.info.vram = vramGB
   }

package/package.json

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

package/test/sysinfo-gpu-driver.test.js

@@ -0,0 +1,87 @@
+const assert = require('node:assert/strict')
+const test = require('node:test')
+
+const system = require('systeminformation')
+const Sysinfo = require('../kernel/sysinfo')
+
+test('GPU sysinfo exposes per-controller drivers and primary NVIDIA driver', async (t) => {
+  t.mock.method(system, 'graphics', async () => ({
+    controllers: [
+      {
+        vendor: 'Intel',
+        model: 'Intel Arc A770',
+        vram: 8192,
+        driverVersion: '31.0.101.5590'
+      },
+      {
+        vendor: 'NVIDIA',
+        model: 'NVIDIA RTX A4500',
+        vram: 20480,
+        driverVersion: '565.90'
+      }
+    ],
+    displays: []
+  }))
+
+  const sys = new Sysinfo()
+  sys.info = {}
+
+  await sys.gpus()
+
+  assert.equal(sys.info.gpu, 'nvidia')
+  assert.equal(sys.info.gpu_model, 'nvidia rtx a4500')
+  assert.equal(sys.info.gpu_driver, '565.90')
+  assert.deepEqual(sys.info.gpus, [
+    {
+      name: 'intel',
+      model: 'intel arc a770',
+      driver: '31.0.101.5590'
+    },
+    {
+      name: 'nvidia',
+      model: 'nvidia rtx a4500',
+      driver: '565.90'
+    }
+  ])
+})
+
+test('GPU sysinfo uses selected highest-VRAM AMD controller for gpu_driver', async (t) => {
+  t.mock.method(system, 'graphics', async () => ({
+    controllers: [
+      {
+        vendor: 'Advanced Micro Devices',
+        model: 'AMD Radeon RX 7600',
+        vram: 8192,
+        driverVersion: '31.0.1'
+      },
+      {
+        vendor: 'AMD',
+        model: 'AMD Radeon RX 7900 XTX',
+        vram: 24576,
+        driverVersion: '31.0.2'
+      }
+    ],
+    displays: []
+  }))
+
+  const sys = new Sysinfo()
+  sys.info = {}
+
+  await sys.gpus()
+
+  assert.equal(sys.info.gpu, 'amd')
+  assert.equal(sys.info.gpu_model, 'amd radeon rx 7900 xtx')
+  assert.equal(sys.info.gpu_driver, '31.0.2')
+  assert.deepEqual(sys.info.gpus, [
+    {
+      name: 'advanced micro devices',
+      model: 'amd radeon rx 7600',
+      driver: '31.0.1'
+    },
+    {
+      name: 'amd',
+      model: 'amd radeon rx 7900 xtx',
+      driver: '31.0.2'
+    }
+  ])
+})