[js/node] gpuMemLimit as double then cast to u64

microsoft · Jan 2, 2025 · 720abb3 · 720abb3
1 parent e03f545
commit 720abb3
Showing 1 changed file with 1 addition and 1 deletion.
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::Number>().DoubleValue());
       }
 #endif
 #ifdef USE_COREML