From 83905c65c759a6e2e2b87d219be3706e74976084 Mon Sep 17 00:00:00 2001 From: josc146 Date: Tue, 30 May 2023 22:46:53 +0800 Subject: [PATCH] tesla P104 P106 kernel --- frontend/src/utils/index.tsx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx index fc882bb..62b8d2f 100644 --- a/frontend/src/utils/index.tsx +++ b/frontend/src/utils/index.tsx @@ -298,7 +298,7 @@ export function toastWithButton(text: string, buttonText: string, onClickButton: } 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'; else if ([' 40'].some(v => commonStore.status.device_name.includes(v))) return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd';