path: root/js/src/wasm/WasmBuiltinModule.yaml

# The file contains information need to define wasm builtin module funcs
# operations.

# i8vecmul(dest: i32, src1: i32, src2: i32, len: i32)
#  Performs pairwise multiplication of two i8 vectors of 'len' specified at
#  'src1' and 'src2'. Output is written to 'dest'. This is used as a
#  basic self-test for builtin module funcs.
- op: I8VecMul
  symbolic_address:
    name: IntrI8VecMul
    type: Args_Int32_GeneralInt32Int32Int32Int32General
  entry: Instance::intrI8VecMul
  export: i8vecmul
  params:
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true

#if defined(ENABLE_WASM_MOZ_INTGEMM)

# Intrinsics for integer matrix multiplication followed by addition of bias.
# Please refer to @TOPSRCDIR/js/src/intgemm/IntegerGemmIntrinsic.h for more details on these intrinsics.


# Prepare B for the Matrix Multiply intrinsic from Input matrix B.
#
#  Quantization is performed on the input.
#  The final prepared B is in CPU-dependent format and can be used as an input to matrix multiply
#  intrinsic (`int8_multiply_and_add_bias`).
#
# int8_prepare_b(const float* inputMatrixB, float scale, float zeroPoint, uint32_t rowsB, uint32_t colsB, int8_t* outputMatrixB)
# int8_prepare_b(inputMatrixB: i32, scale: f32, zeroPoint: f32, rowsB: i32, colsB: i32, outputMatrixB: i32)
- op: I8PrepareB
  symbolic_address:
    name: IntrI8PrepareB
    type: Args_Int32_GeneralInt32Float32Float32Int32Int32Int32General
  entry: intgemm::IntrI8PrepareB
  export: int8_prepare_b
  params:
    - 'i32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Prepare B for the Matrix Multiply intrinsic from transposed version of Input matrix B.
#
#  Quantization is performed on floating values of input.
#  The final prepared B is in CPU-dependent format and can be used as an input to matrix multiply
#  intrinsic (`int8_multiply_and_add_bias`).
#
# int8_prepare_b_from_transposed(const float* inputMatrixBTransposed, float scale, float zeroPoint, uint32_t rowsB, uint32_t colsB, int8_t* outputMatrixB)
# int8_prepare_b_from_transposed(inputMatrixBTransposed: i32, scale: f32, zeroPoint: f32, rowsB: i32, colsB: i32, outputMatrixB: i32)
- op: I8PrepareBFromTransposed
  symbolic_address:
    name: IntrI8PrepareBFromTransposed
    type: Args_Int32_GeneralInt32Float32Float32Int32Int32Int32General
  entry: intgemm::IntrI8PrepareBFromTransposed
  export: int8_prepare_b_from_transposed
  params:
    - 'i32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Prepare B for the Matrix Multiply intrinsic from a quantized and transposed version of Input
# matrix B which is also in a CPU-independent format.
#
#  The final prepared B is in CPU-dependent format and can be used as an input to matrix multiply
#  intrinsic (`int8_multiply_and_add_bias`).
#
# int8_prepare_b_from_quantized_transposed(const int8_t* inputMatrixBQuantizedTransposed, uint32_t rowsB, uint32_t colsB, int8_t* outputMatrixB)
# int8_prepare_b_from_quantized_transposed(inputMatrixBQuantizedTransposed: i32, rowsB: i32, colsB: i32, outputMatrixB: i32)
- op: I8PrepareBFromQuantizedTransposed
  symbolic_address:
    name: IntrI8PrepareBFromQuantizedTransposed
    type: Args_Int32_GeneralInt32Int32Int32Int32General
  entry: intgemm::IntrI8PrepareBFromQuantizedTransposed
  export: int8_prepare_b_from_quantized_transposed
  params:
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Prepare A for the Matrix Multiply intrinsic from Input matrix A.
#
#  It performs quantization on floating values of input.
#  The final prepared A might be architecture dependent. e.g. On some architectures like x86, it
#  might be unsigned (achieved by adding 127 to quantized values) while on others like Arm, it might
#  be signed.
#  The final prepared A can be used as an input to matrix multiply intrinsic
#  (`int8_multiply_and_add_bias`).
#
# int8_prepare_a(const float* inputMatrixA, float scale, float zeroPoint, uint32_t rowsA, uint32_t colsA, int8_t* outputMatrixA)
# int8_prepare_a(inputMatrixA: i32, scale: f32, zeroPoint: f32, rowsA: i32, colsA: i32, outputMatrixA: i32)
- op: I8PrepareA
  symbolic_address:
    name: IntrI8PrepareA
    type: Args_Int32_GeneralInt32Float32Float32Int32Int32Int32General
  entry: intgemm::IntrI8PrepareA
  export: int8_prepare_a
  params:
    - 'i32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Prepares bias for the Matrix Multiply intrinsic.
#
#  It uses the prepared B (which must be obtained by using any of the `int8_prepare_b*` intrinsics) and
#  a bias input to prepare the final bias.
#
#  The final bias can be used as an input to matrix multiply intrinsic (`int8_multiply_and_add_bias`).
#
# int8_prepare_bias(const int8_t* inputMatrixBPrepared, float scaleA, float zeroPointA, float scaleB, float zeroPointB, uint32_t rowsB, uint32_t colsB, const float* inputBias, float* output)
# int8_prepare_bias(inputMatrixBPrepared: i32, scaleA: f32, zeroPointA: f32, scaleB: f32, zeroPointB: f32, rowsB: i32, colsB: i32, inputBias: i32, output: i32)
- op: I8PrepareBias
  symbolic_address:
    name: IntrI8PrepareBias
    type: Args_Int32_GeneralInt32Float32Float32Float32Float32Int32Int32Int32Int32General
  entry: intgemm::IntrI8PrepareBias
  export: int8_prepare_bias
  params:
    - 'i32'
    - 'f32'
    - 'f32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Perform multiplication of 2 matrices followed by adding a bias.
#
#  i.e Output = inputMatrixAPrepared * inputMatrixBPrepared + inputBiasPrepared
#
#  The inputs of this intrinsic must be obtained by using `int8_prepare_A`,
#  one of the `int8_prepare_b*` and `int8_prepare_bias` intrinsics respectively.
#
# int8_multiply_and_add_bias(const int8_t* inputMatrixAPrepared, float scaleA, float zeroPointA,
#                      const int8_t* inputMatrixBPrepared, float scaleB, float zeroPointB,
#                      const float* inputBiasPrepared, float unquantMultiplier,
#                      uint32_t rowsA, uint32_t width, uint32_t colsB, float* output)
# int8_multiply_and_add_bias(inputMatrixAPrepared: i32, scaleA: f32, zeroPointA: f32,
#                      inputMatrixBPrepared: i32, scaleB: f32, zeroPointB: f32,
#                      inputBiasPrepared: i32, unquantMultiplier: f32,
#                      rowsA: i32, width: i32, colsB: i32, output: i32)
- op: I8MultiplyAndAddBias
  symbolic_address:
    name: IntrI8MultiplyAndAddBias
    type: Args_Int32_GeneralInt32Float32Float32Int32Float32Float32Int32Float32Int32Int32Int32Int32General
  entry: intgemm::IntrI8MultiplyAndAddBias
  export: int8_multiply_and_add_bias
  params:
    - 'i32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'f32'
    - 'f32'
    - 'i32'
    - 'f32'
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true


# Select a subset of columns of prepared B.
#
#  Indices of the columns to be selected are specified by an array.
#
# int8_select_columns_of_b(const int8_t* inputMatrixBPrepared, uint32_t rowsB, uint32_t colsB, const uint32_t* colIndexList, const uint32_t sizeColIndexList, int8_t* output)
# int8_select_columns_of_b(inputMatrixBPrepared: i32, rowsB: i32, colsB: i32, colIndexList: i32, sizeColIndexList: i32, output: i32)
- op: I8SelectColumnsOfB
  symbolic_address:
    name: IntrI8SelectColumnsOfB
    type: Args_Int32_GeneralInt32Int32Int32Int32Int32Int32General
  entry: intgemm::IntrI8SelectColumnsOfB
  export: int8_select_columns_of_b
  params:
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
    - 'i32'
  fail_mode: FailOnNegI32
  uses_memory: true

#endif // ENABLE_WASM_MOZ_INTGEMM

#if defined(ENABLE_WASM_JS_STRING_BUILTINS)

- op: StringTest
  symbolic_address:
    name: StringTest
    type: Args_Int32_GeneralGeneral
  entry: Instance::stringTest
  export: test
  params:
    - 'externref'
  result: 'i32'
  fail_mode: Infallible
  uses_memory: false

- op: StringCast
  symbolic_address:
    name: StringCast
    type: Args_General2
  entry: Instance::stringCast
  export: cast
  params:
    - 'externref'
  result:
    code: 'extern'
    nullable: false
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringFromCharCodeArray
  symbolic_address:
    name: StringFromCharCodeArray
    type: Args_General_GeneralGeneralInt32Int32
  entry: Instance::stringFromCharCodeArray
  export: fromCharCodeArray
  params:
    - type: "StaticTypeDefs::arrayMutI16"
      nullable: true
    - 'i32'
    - 'i32'
  result:
    code: 'extern'
    nullable: false
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringIntoCharCodeArray
  symbolic_address:
    name: StringIntoCharCodeArray
    type: Args_Int32_GeneralGeneralGeneralInt32
  entry: Instance::stringIntoCharCodeArray
  export: intoCharCodeArray
  params:
    - 'externref'
    - type: "StaticTypeDefs::arrayMutI16"
      nullable: true
    - 'i32'
  result: 'i32'
  fail_mode: FailOnNegI32
  uses_memory: false

- op: StringFromCharCode
  symbolic_address:
    name: StringFromCharCode
    type: Args_General_GeneralInt32
  entry: Instance::stringFromCharCode
  export: fromCharCode
  params:
    - 'i32'
  result: 'externref'
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringFromCodePoint
  symbolic_address:
    name: StringFromCodePoint
    type: Args_General_GeneralInt32
  entry: Instance::stringFromCodePoint
  export: fromCodePoint
  params:
    - 'i32'
  result: 'externref'
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringCharCodeAt
  symbolic_address:
    name: StringCharCodeAt
    type: Args_Int32_GeneralGeneralInt32
  entry: Instance::stringCharCodeAt
  export: charCodeAt
  params:
    - 'externref'
    - 'i32'
  result: 'i32'
  fail_mode: FailOnNegI32
  uses_memory: false

- op: StringCodePointAt
  symbolic_address:
    name: StringCodePointAt
    type: Args_Int32_GeneralGeneralInt32
  entry: Instance::stringCodePointAt
  export: codePointAt
  params:
    - 'externref'
    - 'i32'
  result: 'i32'
  fail_mode: FailOnNegI32
  uses_memory: false

- op: StringLength
  symbolic_address:
    name: StringLength
    type: Args_Int32_GeneralGeneral
  entry: Instance::stringLength
  export: length
  params:
    - 'externref'
  result: 'i32'
  fail_mode: FailOnNegI32
  uses_memory: false

- op: StringConcat
  symbolic_address:
    name: StringConcat
    type: Args_General3
  entry: Instance::stringConcat
  export: concat
  params:
    - 'externref'
    - 'externref'
  result: 'externref'
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringSubstring
  symbolic_address:
    name: StringSubstring
    type: Args_General_GeneralGeneralInt32Int32
  entry: Instance::stringSubstring
  export: substring
  params:
    - 'externref'
    - 'i32'
    - 'i32'
  result: 'externref'
  fail_mode: FailOnNullPtr
  uses_memory: false

- op: StringEquals
  symbolic_address:
    name: StringEquals
    type: Args_Int32_GeneralGeneralGeneral
  entry: Instance::stringEquals
  export: equals
  params:
    - 'externref'
    - 'externref'
  result: 'i32'
  fail_mode: FailOnNegI32
  uses_memory: false

- op: StringCompare
  symbolic_address:
    name: StringCompare
    type: Args_Int32_GeneralGeneralGeneral
  entry: Instance::stringCompare
  export: compare
  params:
    - 'externref'
    - 'externref'
  result: 'i32'
  fail_mode: FailOnMaxI32
  uses_memory: false

#endif // ENABLE_WASM_JS_STRING_BUILTINS