onnxruntime-node 1.17.3 → 1.18.0

package/dist/version.js

@@ -5,5 +5,5 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.version = void 0;
 // This file is generated by /js/scripts/update-version.ts
 // Do not modify file content manually.
-exports.version = '1.17.3';
+exports.version = '1.18.0';
 //# sourceMappingURL=version.js.map

package/lib/version.ts

@@ -4,4 +4,4 @@
 // This file is generated by /js/scripts/update-version.ts
 // Do not modify file content manually.
 
-export const version = '1.17.3';
+export const version = '1.18.0';

package/package.json

@@ -13,9 +13,9 @@
       3
     ]
   },
-  "version": "1.17.3",
+  "version": "1.18.0",
   "dependencies": {
-    "onnxruntime-common": "1.17.3",
+    "onnxruntime-common": "1.18.0",
     "tar": "^7.0.1"
   },
   "scripts": {

package/script/build.js

@@ -47,6 +47,8 @@ if (ARCH !== 'x64' && ARCH !== 'ia32' && ARCH !== 'arm64' && ARCH !== 'arm') {
 }
 // --onnxruntime-build-dir=
 const ONNXRUNTIME_BUILD_DIR = buildArgs['onnxruntime-build-dir'];
+// --onnxruntime-generator=
+const ONNXRUNTIME_GENERATOR = buildArgs['onnxruntime-generator'];
 // --rebuild
 const REBUILD = !!buildArgs.rebuild;
 // --use_dml
@@ -76,6 +78,9 @@ const args = [
 if (ONNXRUNTIME_BUILD_DIR && typeof ONNXRUNTIME_BUILD_DIR === 'string') {
     args.push(`--CDONNXRUNTIME_BUILD_DIR=${ONNXRUNTIME_BUILD_DIR}`);
 }
+if (ONNXRUNTIME_GENERATOR && typeof ONNXRUNTIME_GENERATOR === 'string') {
+    args.push(`--CDONNXRUNTIME_GENERATOR=${ONNXRUNTIME_GENERATOR}`);
+}
 if (USE_DML) {
     args.push('--CDUSE_DML=ON');
 }

package/script/build.ts

@@ -23,6 +23,8 @@ if (ARCH !== 'x64' && ARCH !== 'ia32' && ARCH !== 'arm64' && ARCH !== 'arm') {
 }
 // --onnxruntime-build-dir=
 const ONNXRUNTIME_BUILD_DIR = buildArgs['onnxruntime-build-dir'];
+// --onnxruntime-generator=
+const ONNXRUNTIME_GENERATOR = buildArgs['onnxruntime-generator'];
 // --rebuild
 const REBUILD = !!buildArgs.rebuild;
 // --use_dml
@@ -55,6 +57,9 @@ const args = [
 if (ONNXRUNTIME_BUILD_DIR && typeof ONNXRUNTIME_BUILD_DIR === 'string') {
   args.push(`--CDONNXRUNTIME_BUILD_DIR=${ONNXRUNTIME_BUILD_DIR}`);
 }
+if (ONNXRUNTIME_GENERATOR && typeof ONNXRUNTIME_GENERATOR === 'string') {
+  args.push(`--CDONNXRUNTIME_GENERATOR=${ONNXRUNTIME_GENERATOR}`);
+}
 if (USE_DML) {
   args.push('--CDUSE_DML=ON');
 }

package/script/install.js

@@ -29,6 +29,8 @@ const {Readable} = require('stream');
 // --onnxruntime-node-install-cuda=v12     Force install the CUDA EP binaries for CUDA 12.
 // --onnxruntime-node-install-cuda=skip    Skip the installation of the CUDA EP binaries.
 //
+// Alternatively, use environment variable "ONNXRUNTIME_NODE_INSTALL_CUDA"
+//
 // If the flag is not provided, the script will only install the CUDA EP binaries when:
 // - The platform is Linux/x64.
 // - The binaries are not already present in the package.
@@ -58,7 +60,7 @@ if (NO_INSTALL || !shouldInstall) {
 const artifactUrl = {
   11: `https://github.com/microsoft/onnxruntime/releases/download/v${ORT_VERSION}/onnxruntime-linux-x64-gpu-${
       ORT_VERSION}.tgz`,
-  12: `https://github.com/microsoft/onnxruntime/releases/download/v${ORT_VERSION}/onnxruntime-linux-x64-cuda12-${
+  12: `https://github.com/microsoft/onnxruntime/releases/download/v${ORT_VERSION}/onnxruntime-linux-x64-gpu-cuda12-${
       ORT_VERSION}.tgz`
 }[INSTALL_CUDA_FLAG || tryGetCudaVersion()];
 console.log(`Downloading "${artifactUrl}"...`);
@@ -79,15 +81,19 @@ Use "--onnxruntime-node-install-cuda=skip" to skip the installation. You will st
   ]);
 
   Readable.fromWeb(res.body)
-      .pipe(tar.t())
-      .on('entry',
-          (entry) => {
-            const filename = path.basename(entry.path);
-            if (entry.type === 'File' && FILES.has(filename)) {
-              console.log(`Extracting "${filename}" to "${BIN_FOLDER}"...`);
-              entry.pipe(fs.createWriteStream(path.join(BIN_FOLDER, filename)));
-            }
-          })
+      .pipe(tar.t({
+        strict: true,
+        onentry: (entry) => {
+          const filename = path.basename(entry.path);
+          if (entry.type === 'File' && FILES.has(filename)) {
+            console.log(`Extracting "${filename}" to "${BIN_FOLDER}"...`);
+            entry.pipe(fs.createWriteStream(path.join(BIN_FOLDER, filename)));
+            entry.on('finish', () => {
+              console.log(`Finished extracting "${filename}".`);
+            });
+          }
+        }
+      }))
       .on('error', (err) => {
         throw new Error(`Failed to extract the binaries: ${err.message}.
 
@@ -105,7 +111,7 @@ function tryGetCudaVersion() {
 }
 
 function parseInstallCudaFlag() {
-  let flag = process.env.npm_config_onnxruntime_node_install_cuda;
+  let flag = process.env.ONNXRUNTIME_NODE_INSTALL_CUDA || process.env.npm_config_onnxruntime_node_install_cuda;
   if (!flag) {
     for (let i = 0; i < process.argv.length; i++) {
       if (process.argv[i].startsWith('--onnxruntime-node-install-cuda=')) {
@@ -118,6 +124,8 @@ function parseInstallCudaFlag() {
   }
   switch (flag) {
     case 'true':
+    case '1':
+    case 'ON':
       return tryGetCudaVersion();
     case 'v11':
       return 11;