Initial commit

Author: jdalton

.github/actions/restore-checkpoint/action.yml

@@ -52,12 +52,16 @@ runs:
         MODE_CHECKPOINT_DIR="packages/${PACKAGE_NAME}/build/${BUILD_MODE}/checkpoints"
         SHARED_CHECKPOINT_DIR="packages/${PACKAGE_NAME}/build/shared/checkpoints"
         OUTPUT_DIR="packages/${PACKAGE_NAME}/build/${BUILD_MODE}/out"
+        SHARED_SOURCE_DIR="packages/${PACKAGE_NAME}/build/shared/source"
+        MODE_SOURCE_DIR="packages/${PACKAGE_NAME}/build/${BUILD_MODE}/source"
 
         echo "📦 Package: ${PACKAGE_NAME}"
         echo "🔧 Build mode: ${BUILD_MODE}"
         echo "📁 Mode checkpoint directory: ${MODE_CHECKPOINT_DIR}"
         echo "📁 Shared checkpoint directory: ${SHARED_CHECKPOINT_DIR}"
         echo "📤 Output directory: ${OUTPUT_DIR}"
+        echo "📂 Shared source directory: ${SHARED_SOURCE_DIR}"
+        echo "📂 Mode source directory: ${MODE_SOURCE_DIR}"
         echo ""
 
         # Parse checkpoint chain into array (comma-separated)
@@ -149,10 +153,82 @@ runs:
         fi
         echo ""
 
-        # Extract checkpoint
-        echo "📦 Extracting checkpoint to ${OUTPUT_DIR}..."
-        mkdir -p "${OUTPUT_DIR}"
-        tar -xzf "${CHECKPOINT_FILE}" -C "${OUTPUT_DIR}"
+        # Extract checkpoint to correct directory based on checkpoint name
+        # Checkpoint structure varies by package:
+        #
+        # node-smol-builder:
+        # - finalized → build/{mode}/out/Final/
+        # - binary-compressed → build/{mode}/out/Compressed/
+        # - binary-stripped → build/{mode}/out/Stripped/
+        # - binary-released → build/{mode}/out/Release/
+        # - source-cloned → build/shared/source/
+        # - source-patched → build/{mode}/source/
+        #
+        # onnxruntime-builder, yoga-layout-builder:
+        # - finalized → build/{mode}/out/Final/
+        # - wasm-synced → build/{mode}/out/Synced/
+        # - wasm-released → build/{mode}/out/Released/
+        # - wasm-optimized → build/{mode}/out/Optimized/
+        # - wasm-compiled → build/{mode}/out/Compiled/
+        # - source-configured → build/{mode}/source/ (yoga-layout only)
+        # - source-cloned → build/shared/source/
+        #
+        # models:
+        # - finalized → build/{mode}/out/Final/
+        # - quantized → build/{mode}/models/
+        # - converted → build/{mode}/models/
+        # - downloaded → build/{mode}/models/
+        EXTRACT_DIR="${OUTPUT_DIR}"
+        case "${RESTORED_CHECKPOINT}" in
+          finalized)
+            EXTRACT_DIR="${OUTPUT_DIR}/Final"
+            ;;
+          # node-smol-builder checkpoints
+          binary-compressed)
+            EXTRACT_DIR="${OUTPUT_DIR}/Compressed"
+            ;;
+          binary-stripped)
+            EXTRACT_DIR="${OUTPUT_DIR}/Stripped"
+            ;;
+          binary-released)
+            EXTRACT_DIR="${OUTPUT_DIR}/Release"
+            ;;
+          # onnxruntime-builder, yoga-layout-builder checkpoints
+          wasm-synced)
+            EXTRACT_DIR="${OUTPUT_DIR}/Synced"
+            ;;
+          wasm-released)
+            EXTRACT_DIR="${OUTPUT_DIR}/Released"
+            ;;
+          wasm-optimized)
+            EXTRACT_DIR="${OUTPUT_DIR}/Optimized"
+            ;;
+          wasm-compiled)
+            EXTRACT_DIR="${OUTPUT_DIR}/Compiled"
+            ;;
+          source-configured)
+            # yoga-layout-builder specific: extract to mode source directory
+            EXTRACT_DIR="${MODE_SOURCE_DIR}"
+            ;;
+          # models checkpoints
+          quantized|converted|downloaded)
+            # Models checkpoints extract to build/{mode}/models/
+            EXTRACT_DIR="packages/${PACKAGE_NAME}/build/${BUILD_MODE}/models"
+            ;;
+          # Source checkpoints (shared across packages)
+          source-cloned)
+            # Shared checkpoint: extract to shared source directory
+            EXTRACT_DIR="${SHARED_SOURCE_DIR}"
+            ;;
+          source-patched)
+            # Mode-specific checkpoint: extract to mode source directory
+            EXTRACT_DIR="${MODE_SOURCE_DIR}"
+            ;;
+        esac
+
+        echo "📦 Extracting checkpoint to ${EXTRACT_DIR}..."
+        mkdir -p "${EXTRACT_DIR}"
+        tar -xzf "${CHECKPOINT_FILE}" -C "${EXTRACT_DIR}"
         echo "✅ Checkpoint extracted successfully"
         echo ""
 

packages/node-smol-builder/scripts/binary-released/shared/build-release.mjs

@@ -94,11 +94,13 @@ const logger = getDefaultLogger()
  * @param {boolean} options.autoYes - Auto-yes to prompts
  * @param {boolean} options.isCI - Whether running in CI
  * @param {boolean} options.isProdBuild - Whether this is a production build
+ * @param {boolean} options.allowCross - Allow cross-compilation (experimental)
  * @param {function} options.collectBuildSourceFiles - Function to collect build source files
  * @param {string} options.packageRoot - Package root directory
  */
 export async function buildRelease(options) {
   const {
+    allowCross,
     arch,
     autoYes,
     bootstrapFile,
@@ -900,18 +902,27 @@ export async function buildRelease(options) {
     }
   }
 
-  // Cross-compilation support (Windows only).
+  // Cross-compilation support (Windows only, or with --allow-cross flag).
   const hostArch = process.arch
   const isArchMismatch = arch !== hostArch
-  const isCrossCompiling = isArchMismatch && IS_WINDOWS
+  const isCrossCompiling = isArchMismatch && (IS_WINDOWS || allowCross)
   if (isCrossCompiling) {
-    logger.log(`Cross-compiling for Windows ${arch} on ${hostArch} host`)
+    if (IS_WINDOWS) {
+      logger.log(`Cross-compiling for Windows ${arch} on ${hostArch} host`)
+    } else {
+      logger.warn(
+        `Cross-compiling for ${platform} ${arch} on ${hostArch} host (experimental)`,
+      )
+      logger.warn(
+        '   This may cause build errors. Use a native runner if possible.',
+      )
+    }
     configureFlags.push(`--dest-cpu=${arch}`)
   } else if (isArchMismatch) {
     logger.fail(
       `Cross-compilation not supported: building ${arch} on ${hostArch} host`,
     )
-    logger.log(`   Use a native ${arch} runner instead.`)
+    logger.log(`   Use a native ${arch} runner or add --allow-cross flag.`)
     process.exit(1)
   }
 

packages/node-smol-builder/scripts/common/shared/build.mjs

@@ -108,6 +108,7 @@ const logger = getDefaultLogger()
 // Parse arguments.
 const { values } = parseArgs({
   options: {
+    'allow-cross': { type: 'boolean', short: 'X' },
     arch: { type: 'string' },
     clean: { type: 'boolean' },
     dev: { type: 'boolean' },
@@ -323,6 +324,7 @@ async function main() {
     autoYes: AUTO_YES,
     isCI: IS_CI,
     isProdBuild: IS_PROD_BUILD,
+    allowCross: !!values['allow-cross'],
     collectBuildSourceFiles: () => collectBuildSourceFiles('binary-released'),
     packageRoot: PACKAGE_ROOT,
   })