add customCudaFile support

2023-05-23 14:04:06 +08:00
parent 65d92d5da1
commit 4eca1537a7
6 changed files with 56 additions and 3 deletions
--- a/frontend/src/utils/index.tsx
+++ b/frontend/src/utils/index.tsx
@@ -267,4 +267,13 @@ export function toastWithButton(text: string, buttonText: string, onClickButton:
      type: 'info',
      ...options
    });
+}
+
+export function getSupportedCustomCudaFile() {
+  if ([' 10', ' 20', ' 30'].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';
+  else
+    return '';
 }