[js/node] gpuMemLimit only useful to be bigint

microsoft · Dec 31, 2024 · f8cf9a3 · f8cf9a3
1 parent e03f545
commit f8cf9a3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/js/common/lib/inference-session.ts b/js/common/lib/inference-session.ts
@@ -223,7 +223,7 @@ export declare namespace InferenceSession {
   export interface CudaExecutionProviderOption extends ExecutionProviderOption {
     readonly name: 'cuda';
     deviceId?: number;
-    gpuMemLimit?: number;
+    gpuMemLimit?: bigint;
 
     /**
      * Arena extend strategy. See

diff --git a/js/node/src/session_options_helper.cc b/js/node/src/session_options_helper.cc
@@ -70,7 +70,7 @@ void ParseExecutionProviders(const Napi::Array epList, Ort::SessionOptions& sess
       }
       if (obj.Has("gpuMemLimit")) {
         gpuMemLimit = static_cast<size_t>(
-            obj.Get("gpuMemLimit").As<Napi::Number>().Uint32Value());
+            obj.Get("gpuMemLimit").As<Napi::BigInt>().Uint64Value());
       }
 #endif
 #ifdef USE_COREML