theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tesla P104 P106 kernel

Browse Source
This commit is contained in:
josc146 2023-05-30 22:46:53 +08:00
parent 9af2b1d208
commit 83905c65c7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontend/src/utils/index.tsx
View File

@ -298,7 +298,7 @@ export function toastWithButton(text: string, buttonText: string, onClickButton:
} }
export function getSupportedCustomCudaFile() { export function getSupportedCustomCudaFile() {
if ([' 10', ' 16', ' 20', ' 30', 'P40'].some(v => commonStore.status.device_name.includes(v))) if ([' 10', ' 16', ' 20', ' 30', 'P40', 'P104', 'P106'].some(v => commonStore.status.device_name.includes(v)))
return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd'; return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd';
else if ([' 40'].some(v => commonStore.status.device_name.includes(v))) else if ([' 40'].some(v => commonStore.status.device_name.includes(v)))
return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd'; return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd';