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

support for tesla P40 custom cuda kernel

Browse Source
This commit is contained in:
josc146 2023-05-29 20:15:37 +08:00
parent 55bb33bcbb
commit 5e698a8312
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

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