index.js

export {
  default as init,
  initSync,
  Model,
  Tensor,
} from "./dist/rten.js";

/**
 * Return true if the current JS environment supports the SIMD extension for
 * WebAssembly.
 */
function simdSupported() {
  // Tiny WebAssembly file generated from the following source using `wat2wasm`:
  //
  // (module
  //   (func (result v128)
  //     i32.const 0
  //     i8x16.splat
  //     i8x16.popcnt
  //   )
  // )
  const simdTest = Uint8Array.from([
    0, 97, 115, 109, 1, 0, 0, 0, 1, 5, 1, 96, 0, 1, 123, 3, 2, 1, 0, 10, 10, 1,
    8, 0, 65, 0, 253, 15, 253, 98, 11,
  ]);
  return WebAssembly.validate(simdTest);
}

/**
 * Return the filename of the preferred RTen binary for the current
 * environment.
 */
export function binaryName() {
  if (simdSupported()) {
    return "rten_bg.wasm";
  } else {
    return "rten-nosimd_bg.wasm";
  }
}