Fix storing unboxed_vec128 in closures

Author: Ekdohibs

middle_end/clambda_layout.ml

@@ -17,6 +17,7 @@ type atom =
   | Value_int
   | Unboxed_float
   | Unboxed_int of Lambda.boxed_integer
+  | Unboxed_vector of Lambda.boxed_vector
 
 let rec fold_left_layout (f : 'acc -> 'e -> atom -> 'acc) (acc : 'acc)
     (expr : Clambda.ulambda) (layout : Clambda_primitives.layout) : 'acc =
@@ -28,6 +29,7 @@ let rec fold_left_layout (f : 'acc -> 'e -> atom -> 'acc) (acc : 'acc)
        closure."
   | Punboxed_float -> f acc expr Unboxed_float
   | Punboxed_int bi -> f acc expr (Unboxed_int bi)
+  | Punboxed_vector bv -> f acc expr (Unboxed_vector bv)
   | Pvalue Pintval -> f acc expr Value_int
   | Pvalue _ -> f acc expr Value
   | Punboxed_product layouts ->
@@ -59,6 +61,7 @@ let print_atom ppf = function
   | Unboxed_int Pint32 -> Format.fprintf ppf "unboxed_int32"
   | Unboxed_int Pint64 -> Format.fprintf ppf "unboxed_int64"
   | Unboxed_int Pnativeint -> Format.fprintf ppf "unboxed_nativeint"
+  | Unboxed_vector Pvec128 -> Format.fprintf ppf "unboxed_vec128"
 
 let equal_decomposition = ( = )
 
@@ -80,6 +83,7 @@ let rec decompose (layout : Lambda.layout) : _ decomposition' =
        closure."
   | Punboxed_float -> Gc_invisible ((), Unboxed_float)
   | Punboxed_int bi -> Gc_invisible ((), Unboxed_int bi)
+  | Punboxed_vector bv -> Gc_invisible ((), Unboxed_vector bv)
   | Pvalue Pintval -> Gc_invisible ((), Value_int)
   | Pvalue _ -> Gc_visible ((), Value)
   | Punboxed_product l -> Product (Array.of_list (List.map decompose l))
@@ -105,7 +109,9 @@ let rec fold_decompose (f1 : 'acc -> 'a -> atom -> 'acc * 'b)
     acc, Product elts
 
 let atom_size (layout : atom) =
-  match layout with Value | Value_int | Unboxed_float | Unboxed_int _ -> 1
+  match layout with
+  | Value | Value_int | Unboxed_float | Unboxed_int _ -> 1
+  | Unboxed_vector Pvec128 -> 2
 
 let assign_invisible_offsets init_pos (var, dec) =
   let f_visible acc () _layout = acc, () in

middle_end/flambda2/simplify_shared/slot_offsets.ml

@@ -873,10 +873,12 @@ end = struct
           | Region | Rec_info ->
             Misc.fatal_errorf "Value slot %a has Region or Rec_info kind"
               Value_slot.print value_slot
-          | Naked_number _ ->
+          | Naked_number (Naked_immediate | Naked_float | Naked_int32 | Naked_int64 | Naked_nativeint) ->
             1, true
-            (* flambda only supports 64-bits for now, so naked numbers can only
-               be of size 1 *)
+          (* flambda only supports 64-bits for now, so naked numbers can only
+             be of size 1 *)
+          | Naked_number Naked_vec128 ->
+            2, true
           | Value -> (
             match[@ocaml.warning "-4"]
               Flambda_kind.With_subkind.subkind kind