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

fix: multiple gpu (#4641)

Browse Source 
* fix: multiple gpu

* chore: check vulkan when app load

* chore: remove log

* chore: update engine variant conditional
This commit is contained in:
Faisal Amir 2025-02-12 22:50:56 +07:00 committed by GitHub
parent 6d73f49cdf
commit aed42edb3f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 28 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
extensions/engine-management-extension/src/utils.ts
View File

@ -25,10 +25,10 @@ const os = (settings?: GpuSetting): string => {
return PLATFORM === 'win32'
? 'windows-amd64'
: PLATFORM === 'darwin'
? settings?.cpu?.arch === 'arm64'
? 'mac-arm64'
: 'mac-amd64'
: 'linux-amd64'
? settings?.cpu?.arch === 'arm64'
? 'mac-arm64'
: 'mac-amd64'
: 'linux-amd64'
}
/**
@ -65,21 +65,22 @@ export const engineVariant = async (
// There is no need to append the variant extension for mac
if (platform.startsWith('mac')) return platform
let engineVariant = [
platform,
gpuSetting?.vulkan
? 'vulkan'
: (gpuRunMode(gpuSetting) === 'cuda' && // GPU mode - packaged CUDA variants of avx2 and noavx
gpuSetting.cpu.instructions.some((inst) => inst === 'avx2')) ||
gpuSetting.cpu.instructions.some((inst) => inst === 'avx512')
? 'avx2'
: 'noavx',
gpuRunMode(gpuSetting),
cudaVersion(gpuSetting),
]
.filter((e) => !!e)
.join('-')
let engineVariant =
gpuSetting?.vulkan || gpuSetting.gpus.some((e) => !e.additional_information)
? [platform, 'vulkan']
: [
platform,
gpuRunMode(gpuSetting) === 'cuda' &&
(gpuSetting.cpu.instructions.includes('avx2') ||
gpuSetting.cpu.instructions.includes('avx512'))
? 'avx2'
: 'noavx',
gpuRunMode(gpuSetting),
cudaVersion(gpuSetting),
].filter(Boolean) // Remove any falsy values
log(`[CORTEX]: Engine variant: ${engineVariant}`)
return engineVariant
let engineVariantString = engineVariant.join('-')
log(`[CORTEX]: Engine variant: ${engineVariantString}`)
return engineVariantString
}

11
web/screens/Settings/Hardware/index.tsx
View File

@ -50,9 +50,9 @@ const Hardware = () => {
}
// Handle switch toggle for GPU activation
const handleSwitchChange = async (index: number, isActive: boolean) => {
const updatedGpus = gpus.map((gpu, i) =>
i === index ? { ...gpu, activated: isActive } : gpu
const handleSwitchChange = async (id: string, isActive: boolean) => {
const updatedGpus = gpus.map((gpu) =>
gpu.id === id ? { ...gpu, activated: isActive } : gpu
)
setGpus(updatedGpus)
// Call the API to update the active GPUs
@ -282,7 +282,10 @@ const Hardware = () => {
<Switch
checked={item.activated}
onChange={(e) =>
handleSwitchChange(i, e.target.checked)
handleSwitchChange(
item.id,
e.target.checked
)
}
/>