translprim.re

open Grain_parsing;
open Ast_helper;
open Typedtree;
open Parsetree;

type primitive_constant =
  | ElideTypeInfo;

type primitive =
  | PrimitiveConstant(primitive_constant)
  | Primitive0(prim0)
  | Primitive1(prim1)
  | Primitive2(prim2)
  | PrimitiveN(primn);

module PrimMap =
  Hashtbl.Make({
    type t = string;
    let hash = x => Hashtbl.hash(x);
    let equal = (a, b) => String.compare(a, b) === 0;
  });

let default_loc = Location.dummy_loc;

let mkident = name =>
  Expression.ident(
    ~loc=Location.dummy_loc,
    ~core_loc=Location.dummy_loc,
    Location.mkloc(
      Identifier.IdentName(Location.mkloc(name, default_loc)),
      default_loc,
    ),
  );
let mkpatvar = name => {
  ppat_desc: PPatVar(Location.mkloc(name, default_loc)),
  ppat_loc: default_loc,
};

let id_a = mkident("a");
let id_b = mkident("b");
let id_c = mkident("c");
let id_d = mkident("d");
let id_e = mkident("e");
let pat_a = mkpatvar("a");
let pat_b = mkpatvar("b");
let pat_c = mkpatvar("c");
let pat_d = mkpatvar("d");
let pat_e = mkpatvar("e");

let prim_map =
  PrimMap.of_seq(
    List.to_seq([
      ("@heap.start", Primitive0(HeapStart)),
      ("@heap.type_metadata", Primitive0(HeapTypeMetadata)),
      ("@meta.elide_type_info", PrimitiveConstant(ElideTypeInfo)),
      ("@allocate.array", Primitive2(AllocateArray)),
      ("@allocate.tuple", Primitive1(AllocateTuple)),
      ("@allocate.bytes", Primitive1(AllocateBytes)),
      ("@allocate.string", Primitive1(AllocateString)),
      ("@allocate.bigint", Primitive1(AllocateBigInt)),
      ("@allocate.wasm_array_any_ref", Primitive2(AllocateWasmArrayAnyRef)),
      ("@allocate.adt", PrimitiveN(AllocateAdt)),
      ("@allocate.record", PrimitiveN(AllocateRecord)),
      ("@new.int32", Primitive1(NewInt32)),
      ("@new.int64", Primitive1(NewInt64)),
      ("@new.uint32", Primitive1(NewUint32)),
      ("@new.uint64", Primitive1(NewUint64)),
      ("@new.float32", Primitive1(NewFloat32)),
      ("@new.float64", Primitive1(NewFloat64)),
      ("@builtin.id", Primitive1(BuiltinId)),
      ("@record.load_typehash", Primitive1(LoadRecordTypeHash)),
      ("@adt.load_typehash", Primitive1(LoadVariantTypeHash)),
      ("@record.load_typeid", Primitive1(LoadRecordTypeId)),
      ("@adt.load_typeid", Primitive1(LoadVariantTypeId)),
      ("@adt.load_variant", Primitive1(LoadAdtVariant)),
      ("@grain_value.load_tag", Primitive1(LoadValueTag)),
      ("@grain_value.load_cycle_marker", Primitive1(LoadCycleMarker)),
      ("@grain_value.store_cycle_marker", Primitive2(StoreCycleMarker)),
      ("@string.size", Primitive1(StringSize)),
      ("@bytes.size", Primitive1(BytesSize)),
      ("@bigint.size", Primitive1(BigIntSize)),
      ("@bigint.flags", Primitive1(BigIntFlags)),
      ("@string.refarray", Primitive1(StringArrayRef)),
      ("@bytes.refarray", Primitive1(BytesArrayRef)),
      ("@compound_value.refarray", Primitive1(CompoundValueArrayRef)),
      ("@bigint.refarray", Primitive1(BigIntArrayRef)),
      ("@tag.simple_number", Primitive1(TagSimpleNumber)),
      ("@untag.simple_number", Primitive1(UntagSimpleNumber)),
      ("@tag.char", Primitive1(TagChar)),
      ("@untag.char", Primitive1(UntagChar)),
      ("@tag.int8", Primitive1(TagInt8)),
      ("@untag.int8", Primitive1(UntagInt8)),
      ("@tag.int16", Primitive1(TagInt16)),
      ("@untag.int16", Primitive1(UntagInt16)),
      ("@tag.uint8", Primitive1(TagUint8)),
      ("@untag.uint8", Primitive1(UntagUint8)),
      ("@tag.uint16", Primitive1(TagUint16)),
      ("@untag.uint16", Primitive1(UntagUint16)),
      ("@boxed_number.get_tag", Primitive1(BoxedNumberTag)),
      ("@boxed_number.get_int32", Primitive1(BoxedInt32Value)),
      ("@boxed_number.get_uint32", Primitive1(BoxedUint32Value)),
      ("@boxed_number.get_float32", Primitive1(BoxedFloat32Value)),
      ("@boxed_number.get_int64", Primitive1(BoxedInt64Value)),
      ("@boxed_number.get_uint64", Primitive1(BoxedUint64Value)),
      ("@boxed_number.get_float64", Primitive1(BoxedFloat64Value)),
      (
        "@boxed_number.get_rational_numerator",
        Primitive1(BoxedRationalNumerator),
      ),
      (
        "@boxed_number.get_rational_denominator",
        Primitive1(BoxedRationalDenominator),
      ),
      ("@is_refi31", Primitive1(IsRefI31)),
      ("@is_grain_heap_value", Primitive1(IsGrainHeapValue)),
      ("@i31.get_s", Primitive1(I31Get({signed: true}))),
      ("@i31.get_u", Primitive1(I31Get({signed: false}))),
      ("@i31.make", Primitive1(I31Make)),
      ("@not", Primitive1(Not)),
      ("@box", Primitive1(Box)),
      ("@unbox", Primitive1(Unbox)),
      ("@ignore", Primitive1(Ignore)),
      ("@assert", Primitive1(Assert)),
      ("@throw", Primitive1(Throw)),
      ("@magic", Primitive1(Magic)),
      ("@unreachable", Primitive0(Unreachable)),
      ("@is", Primitive2(Is)),
      ("@eq", Primitive2(Eq)),
      ("@and", Primitive2(And)),
      ("@or", Primitive2(Or)),
      ("@array.length", Primitive1(ArrayLength)),
      ("@new.rational", Primitive2(NewRational)),
      ("@bigint.set_flags", Primitive2(BigIntSetFlags)),
      (
        "@wasm.load_int32",
        Primitive2(
          WasmLoadI32({
            sz: 4,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.load_8_s_int32",
        Primitive2(
          WasmLoadI32({
            sz: 1,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.load_8_u_int32",
        Primitive2(
          WasmLoadI32({
            sz: 1,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.load_16_s_int32",
        Primitive2(
          WasmLoadI32({
            sz: 2,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.load_16_u_int32",
        Primitive2(
          WasmLoadI32({
            sz: 2,
            signed: false,
          }),
        ),
      ),
      ("@wasm.store_int32", PrimitiveN(WasmStoreI32({sz: 4}))),
      ("@wasm.store_8_int32", PrimitiveN(WasmStoreI32({sz: 1}))),
      ("@wasm.store_16_int32", PrimitiveN(WasmStoreI32({sz: 2}))),
      (
        "@wasm.load_int64",
        Primitive2(
          WasmLoadI64({
            sz: 8,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.load_8_s_int64",
        Primitive2(
          WasmLoadI64({
            sz: 1,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.load_8_u_int64",
        Primitive2(
          WasmLoadI64({
            sz: 1,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.load_16_s_int64",
        Primitive2(
          WasmLoadI64({
            sz: 2,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.load_16_u_int64",
        Primitive2(
          WasmLoadI64({
            sz: 2,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.load_32_s_int64",
        Primitive2(
          WasmLoadI64({
            sz: 4,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.load_32_u_int64",
        Primitive2(
          WasmLoadI64({
            sz: 4,
            signed: false,
          }),
        ),
      ),
      ("@wasm.store_int64", PrimitiveN(WasmStoreI64({sz: 8}))),
      ("@wasm.store_8_int64", PrimitiveN(WasmStoreI64({sz: 1}))),
      ("@wasm.store_16_int64", PrimitiveN(WasmStoreI64({sz: 2}))),
      ("@wasm.store_32_int64", PrimitiveN(WasmStoreI64({sz: 4}))),
      ("@wasm.load_float32", Primitive2(WasmLoadF32)),
      ("@wasm.store_float32", PrimitiveN(WasmStoreF32)),
      ("@wasm.load_float64", Primitive2(WasmLoadF64)),
      ("@wasm.store_float64", PrimitiveN(WasmStoreF64)),
      (
        "@wasm.clz_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_clz_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.ctz_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_ctz_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.popcnt_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_popcnt_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.eq_z_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_eq_z_int32,
            arg_type: Wasm_int32,
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.add_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_add_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.sub_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_sub_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.mul_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_mul_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.div_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_div_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.div_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_div_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.rem_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_rem_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.rem_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_rem_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.and_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_and_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.or_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_or_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.xor_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_xor_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.shl_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_shl_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.shr_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_shr_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.shr_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_shr_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.rot_l_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_rot_l_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.rot_r_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_rot_r_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.eq_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_eq_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ne_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_ne_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_lt_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_lt_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_le_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_le_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_gt_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_gt_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_s_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_ge_s_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_u_int32",
        Primitive2(
          WasmBinaryI32({
            wasm_op: Op_ge_u_int32,
            arg_types: (Wasm_int32, Wasm_int32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.wrap_int64",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_wrap_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.trunc_s_float32_to_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_trunc_s_float32_to_int32,
            arg_type: Wasm_float32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.trunc_u_float32_to_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_trunc_u_float32_to_int32,
            arg_type: Wasm_float32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.trunc_s_float64_to_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_trunc_s_float64_to_int32,
            arg_type: Wasm_float64,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.trunc_u_float64_to_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_trunc_u_float64_to_int32,
            arg_type: Wasm_float64,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.trunc_u_float64_to_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_trunc_u_float64_to_int32,
            arg_type: Wasm_float64,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.reinterpret_float32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_reinterpret_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.extend_s8_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_extend_s8_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.extend_s16_int32",
        Primitive1(
          WasmUnaryI32({
            wasm_op: Op_extend_s16_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int32,
          }),
        ),
      ),
      (
        "@wasm.clz_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_clz_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.ctz_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_ctz_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.popcnt_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_popcnt_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.eq_z_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_eq_z_int64,
            arg_type: Wasm_int64,
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.add_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_add_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.sub_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_sub_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.mul_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_mul_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.div_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_div_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.div_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_div_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.rem_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_rem_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.rem_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_rem_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.and_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_and_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.or_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_or_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.xor_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_xor_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.shl_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_shl_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.shr_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_shr_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.shr_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_shr_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.rot_l_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_rot_l_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.rot_r_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_rot_r_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.eq_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_eq_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ne_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_ne_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_lt_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_lt_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_le_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_le_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_gt_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_gt_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_s_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_ge_s_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_u_int64",
        Primitive2(
          WasmBinaryI64({
            wasm_op: Op_ge_u_int64,
            arg_types: (Wasm_int64, Wasm_int64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.extend_s_int32",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_extend_s_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.extend_u_int32",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_extend_u_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.trunc_s_float32_to_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_trunc_s_float32_to_int64,
            arg_type: Wasm_float32,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.trunc_u_float32_to_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_trunc_u_float32_to_int64,
            arg_type: Wasm_float32,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.trunc_s_float64_to_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_trunc_s_float64_to_int64,
            arg_type: Wasm_float64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.trunc_u_float64_to_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_trunc_u_float64_to_int64,
            arg_type: Wasm_float64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.reinterpret_float64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_reinterpret_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.extend_s8_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_extend_s8_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.extend_s16_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_extend_s16_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.extend_s32_int64",
        Primitive1(
          WasmUnaryI64({
            wasm_op: Op_extend_s32_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_int64,
          }),
        ),
      ),
      (
        "@wasm.neg_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_neg_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.abs_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_abs_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.ceil_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_ceil_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.floor_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_floor_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.trunc_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_trunc_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.nearest_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_nearest_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.sqrt_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_sqrt_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.add_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_add_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.sub_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_sub_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.mul_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_mul_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.div_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_div_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.copy_sign_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_copy_sign_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.min_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_min_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.max_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_max_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.eq_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_eq_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ne_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_ne_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_lt_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_le_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_gt_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_float32",
        Primitive2(
          WasmBinaryF32({
            wasm_op: Op_ge_float32,
            arg_types: (Wasm_float32, Wasm_float32),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.reinterpret_int32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_reinterpret_int32,
            arg_type: Wasm_int32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.convert_s_int32_to_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_convert_s_int32_to_float32,
            arg_type: Wasm_int32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.convert_u_int32_to_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_convert_u_int32_to_float32,
            arg_type: Wasm_int32,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.convert_s_int64_to_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_convert_s_int64_to_float32,
            arg_type: Wasm_int64,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.convert_u_int64_to_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_convert_u_int64_to_float32,
            arg_type: Wasm_int64,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.demote_float64",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_demote_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float32,
          }),
        ),
      ),
      (
        "@wasm.neg_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_neg_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.abs_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_abs_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.ceil_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_ceil_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.floor_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_floor_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.trunc_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_trunc_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.nearest_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_nearest_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.sqrt_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_sqrt_float64,
            arg_type: Wasm_float64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.add_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_add_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.sub_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_sub_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.mul_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_mul_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.div_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_div_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.copy_sign_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_copy_sign_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.min_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_min_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.max_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_max_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.eq_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_eq_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ne_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_ne_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.lt_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_lt_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.le_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_le_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.gt_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_gt_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.ge_float64",
        Primitive2(
          WasmBinaryF64({
            wasm_op: Op_ge_float64,
            arg_types: (Wasm_float64, Wasm_float64),
            ret_type: Grain_bool,
          }),
        ),
      ),
      (
        "@wasm.reinterpret_int64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_reinterpret_int64,
            arg_type: Wasm_int64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.convert_s_int32_to_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_convert_s_int32_to_float64,
            arg_type: Wasm_int32,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.convert_u_int32_to_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_convert_u_int32_to_float64,
            arg_type: Wasm_int32,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.convert_s_int64_to_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_convert_s_int64_to_float64,
            arg_type: Wasm_int64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.convert_u_int64_to_float64",
        Primitive1(
          WasmUnaryF64({
            wasm_op: Op_convert_u_int64_to_float64,
            arg_type: Wasm_int64,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      (
        "@wasm.promote_float32",
        Primitive1(
          WasmUnaryF32({
            wasm_op: Op_promote_float32,
            arg_type: Wasm_float32,
            ret_type: Wasm_float64,
          }),
        ),
      ),
      ("@wasm.fromGrain", Primitive1(WasmFromGrain)),
      ("@wasm.toGrain", Primitive1(WasmToGrain)),
      ("@wasm.memory_grow", Primitive1(WasmMemoryGrow)),
      ("@wasm.memory_size", Primitive0(WasmMemorySize)),
      ("@wasm.memory_copy", PrimitiveN(WasmMemoryCopy)),
      ("@wasm.memory_fill", PrimitiveN(WasmMemoryFill)),
      ("@wasm.ref_array_len", Primitive1(WasmRefArrayLen)),
      (
        "@wasm.ref_array_i8_get_s",
        Primitive2(
          WasmRefArrayGet({
            array_type: Wasm_packed_i8,
            signed: true,
          }),
        ),
      ),
      (
        "@wasm.ref_array_i8_get_u",
        Primitive2(
          WasmRefArrayGet({
            array_type: Wasm_packed_i8,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.ref_array_i8_set",
        PrimitiveN(WasmRefArraySet({array_type: Wasm_packed_i8})),
      ),
      (
        "@wasm.ref_array_i64_get",
        Primitive2(
          WasmRefArrayGet({
            array_type: Wasm_int64,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.ref_array_i64_set",
        PrimitiveN(WasmRefArraySet({array_type: Wasm_int64})),
      ),
      (
        "@wasm.ref_array_any_get",
        Primitive2(
          WasmRefArrayGet({
            array_type: Wasm_any,
            signed: false,
          }),
        ),
      ),
      (
        "@wasm.ref_array_any_set",
        PrimitiveN(WasmRefArraySet({array_type: Wasm_any})),
      ),
      (
        "@wasm.ref_array_i8_copy",
        PrimitiveN(WasmRefArrayCopy({array_type: Wasm_packed_i8})),
      ),
      (
        "@wasm.ref_array_i64_copy",
        PrimitiveN(WasmRefArrayCopy({array_type: Wasm_int64})),
      ),
      (
        "@wasm.ref_array_any_copy",
        PrimitiveN(WasmRefArrayCopy({array_type: Wasm_any})),
      ),
      (
        "@wasm.ref_array_i8_fill",
        PrimitiveN(WasmRefArrayFill({array_type: Wasm_packed_i8})),
      ),
      (
        "@wasm.ref_array_i64_fill",
        PrimitiveN(WasmRefArrayFill({array_type: Wasm_int64})),
      ),
      (
        "@wasm.ref_array_any_fill",
        PrimitiveN(WasmRefArrayFill({array_type: Wasm_any})),
      ),
    ]),
  );

let active_memory_base = () => {
  switch (Grain_utils.Config.memory_base^) {
  | Some(x) => x
  | None => Grain_utils.Config.default_memory_base
  };
};

let transl_prim = (env, desc) => {
  let loc = desc.pprim_loc;
  let core_loc = desc.pprim_loc;

  let prim =
    try(PrimMap.find(prim_map, desc.pprim_name.txt)) {
    | Not_found => failwith("This primitive does not exist.")
    };

  let lambda_arg = pat => {
    pla_label: Unlabeled,
    pla_pattern: pat,
    pla_default: None,
    pla_loc: Location.dummy_loc,
  };

  let (value, typ) =
    switch (prim) {
    | PrimitiveConstant(const) =>
      let (value, typ) =
        switch (const) {
        | ElideTypeInfo => (
            Constant.bool(Grain_utils.Config.elide_type_info^),
            Builtin_types.type_bool,
          )
        };
      (Expression.constant(~loc, ~core_loc, value), typ);
    | Primitive0(p) => (
        Expression.lambda(
          ~loc,
          ~core_loc,
          [],
          Expression.prim0(~loc, ~core_loc, p),
        ),
        Typecore.prim0_type(p),
      )
    | Primitive1(BuiltinId as p) =>
      // This primitive must always be inlined, so we do not generate a lambda
      (
        Expression.constant(~loc, ~core_loc, PConstVoid),
        Typecore.prim1_type(p),
      )
    | Primitive1(p) => (
        Expression.lambda(
          ~loc,
          ~core_loc,
          [lambda_arg(pat_a)],
          Expression.prim1(~loc, ~core_loc, p, id_a),
        ),
        Typecore.prim1_type(p),
      )
    | Primitive2(p) => (
        Expression.lambda(
          ~loc,
          ~core_loc,
          [lambda_arg(pat_a), lambda_arg(pat_b)],
          Expression.prim2(~loc, ~core_loc, p, id_a, id_b),
        ),
        Typecore.prim2_type(p),
      )
    | PrimitiveN(p) =>
      let (args, values) =
        switch (p) {
        | WasmStoreI32(_)
        | WasmStoreI64(_)
        | WasmStoreF32
        | WasmStoreF64
        | WasmMemoryCopy
        | WasmMemoryFill
        | WasmRefArraySet(_) => (
            [lambda_arg(pat_a), lambda_arg(pat_b), lambda_arg(pat_c)],
            [id_a, id_b, id_c],
          )
        | WasmRefArrayCopy(_) => (
            [
              lambda_arg(pat_a),
              lambda_arg(pat_b),
              lambda_arg(pat_c),
              lambda_arg(pat_d),
              lambda_arg(pat_e),
            ],
            [id_a, id_b, id_c, id_d, id_e],
          )
        | WasmRefArrayFill(_) => (
            [
              lambda_arg(pat_a),
              lambda_arg(pat_b),
              lambda_arg(pat_c),
              lambda_arg(pat_d),
            ],
            [id_a, id_b, id_c, id_d],
          )
        | AllocateRecord => (
            [lambda_arg(pat_a), lambda_arg(pat_b), lambda_arg(pat_c)],
            [id_a, id_b, id_c],
          )
        | AllocateAdt => (
            [
              lambda_arg(pat_a),
              lambda_arg(pat_b),
              lambda_arg(pat_c),
              lambda_arg(pat_d),
            ],
            [id_a, id_b, id_c, id_d],
          )
        };
      (
        Expression.lambda(
          ~loc,
          ~core_loc,
          args,
          Expression.primn(~loc, ~core_loc, p, values),
        ),
        Typecore.primn_type(p),
      );
    };

  let id = Ident.create(desc.pprim_ident.txt);
  let value_description = {
    Types.val_type: typ,
    val_repr: Type_utils.repr_of_type(env, typ),
    val_kind: TValPrim(desc.pprim_name.txt),
    val_loc: loc,
    val_internalpath: PIdent(id),
    val_fullpath: Path.PIdent(id),
    val_mutable: false,
    val_global: true,
  };

  let value = Typecore.type_expression(env, value);
  let env = Env.add_value(id, value_description, env);
  let binds = [
    {
      vb_pat: {
        pat_desc: TPatVar(id, desc.pprim_ident),
        pat_loc: loc,
        pat_extra: [],
        pat_type: typ,
        pat_env: env,
      },
      vb_expr: value,
      vb_loc: loc,
    },
  ];
  (binds, id, value_description, env);
};