From 47e5327228fdb4f92608a7309d500f2f3a823ddb Mon Sep 17 00:00:00 2001 From: Service Account Date: Sat, 21 Oct 2023 18:04:29 -0700 Subject: [PATCH] chore: Add todo for checking whether CUDA exists or not --- plugins/inference-plugin/module.ts | 2 ++ 1 file changed, 2 insertions(+) diff --git a/plugins/inference-plugin/module.ts b/plugins/inference-plugin/module.ts index e3bad81d8..b0318bd03 100644 --- a/plugins/inference-plugin/module.ts +++ b/plugins/inference-plugin/module.ts @@ -33,12 +33,14 @@ const initModel = (fileName) => { let binaryName; if (process.platform === "win32") { + // Todo: Need to check for CUDA support to switch between CUDA and non-CUDA binaries binaryName = "nitro_windows_amd64_cuda.exe"; } else if (process.platform === "darwin") { // Mac OS platform binaryName = process.arch === "arm64" ? "nitro_mac_arm64" : "nitro_mac_intel"; } else { // Linux + // Todo: Need to check for CUDA support to switch between CUDA and non-CUDA binaries binaryName = "nitro_linux_amd64_cuda"; // For other platforms }