Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix feature parsing

Browse Source
This commit is contained in:
Thien Tran 2025-05-30 10:00:22 +08:00 committed by Louis
parent a75d13f42f
commit 27146eb5cc
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/llamacpp-extension/src/backend.ts
View File

@ -154,9 +154,9 @@ export async function downloadBackend(backend: string, version: string): Promise
async function _getSupportedFeatures() { async function _getSupportedFeatures() {
const sysInfo = await window.core.api.getSystemInfo() const sysInfo = await window.core.api.getSystemInfo()
const features = { const features = {
avx: 'avx' in sysInfo.cpu.extensions, avx: sysInfo.cpu.extensions.includes('avx'),
avx2: 'avx2' in sysInfo.cpu.extensions, avx2: sysInfo.cpu.extensions.includes('avx2'),
avx512: 'avx512_f' in sysInfo.cpu.extensions, avx512: sysInfo.cpu.extensions.includes('avx512'),
cuda11: false, cuda11: false,
cuda12: false, cuda12: false,
vulkan: false, vulkan: false,
@ -177,14 +177,14 @@ async function _getSupportedFeatures() {
for (const gpuInfo of sysInfo.gpus) { for (const gpuInfo of sysInfo.gpus) {
const driverVersion = gpuInfo.driver_version const driverVersion = gpuInfo.driver_version
if (gpuInfo.nvidiaInfo?.compute_capability) { if (gpuInfo.nvidia_info?.compute_capability) {
if (compareVersions(driverVersion, minCuda11DriverVersion) >= 0) if (compareVersions(driverVersion, minCuda11DriverVersion) >= 0)
features.cuda11 = true features.cuda11 = true
if (compareVersions(driverVersion, minCuda12DriverVersion) >= 0) if (compareVersions(driverVersion, minCuda12DriverVersion) >= 0)
features.cuda12 = true features.cuda12 = true
} }
if (gpuInfo.vulkanInfo?.api_version) features.vulkan = true if (gpuInfo.vulkan_info?.api_version) features.vulkan = true
} }
return features return features