translmode.ml

(**************************************************************************)
(*                                                                        *)
(*                                 OCaml                                  *)
(*                                                                        *)
(*                    Zesen Qian, Jane Street, London                     *)
(*                                                                        *)
(*   Copyright 2024 Jane Street Group LLC                                 *)
(*                                                                        *)
(*   All rights reserved.  This file is distributed under the terms of    *)
(*   the GNU Lesser General Public License version 2.1, with the          *)
(*   special exception on linking described in the file LICENSE.          *)
(*                                                                        *)
(**************************************************************************)

open Lambda
open Mode

let transl_locality_mode = function
  | Locality.Const.Global -> alloc_heap
  | Locality.Const.Local -> alloc_local

let transl_locality_mode_l locality =
  Locality.zap_to_floor locality
  |> transl_locality_mode

let transl_locality_mode_r locality =
  (* r mode are for allocations; [optimise_allocations] should have pushed it
     to ceil and determined. *)
  Locality.check_const locality
  |> Option.get
  |> transl_locality_mode

let transl_alloc_mode_l mode =
(* we only take the locality axis *)
  Alloc.locality mode
  |> transl_locality_mode_l

let transl_alloc_mode_r mode =
  (* we only take the locality axis *)
  Alloc.locality mode
  |> transl_locality_mode_r

let transl_modify_mode locality =
  match Locality.zap_to_floor locality with
  | Global -> modify_heap
  | Local -> modify_maybe_stack