Source file owl_ext_lifts.ml

# 1 "src/owl/ext/owl_ext_lifts.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * Copyright (c) 2016-2020 Liang Wang <liang.wang@cl.cam.ac.uk>
 *)

(* Currently, the lifting is done by element-by-elment copying. This will be
  very inefficient on large data chunks. The performance of this module needs
  to be improved in the future. *)

open Owl_ext_types

(* lift precision *)

module DAS_DAD = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src = unpack_das

  let pack_dst = pack_dad

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2d src in
    pack_dst dst
end

module DAC_DAZ = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src = unpack_dac

  let pack_dst = pack_daz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_c2z src in
    pack_dst dst
end

module DMS_DMD = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src = unpack_dms

  let pack_dst = pack_dmd

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2d src in
    pack_dst dst
end

module DMC_DMZ = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src = unpack_dmc

  let pack_dst = pack_dmz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_c2z src in
    pack_dst dst
end

(* lift element type *)

module F_C = struct
  let lift src = Complex.{ re = unpack_flt src; im = 0. } |> pack_cpx
end

module DAS_DAC = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src = unpack_das

  let pack_dst = pack_dac

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2c src in
    pack_dst dst
end

module DAD_DAZ = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src = unpack_dad

  let pack_dst = pack_daz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_d2z src in
    pack_dst dst
end

module DMS_DMC = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src = unpack_dms

  let pack_dst = pack_dmc

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2c src in
    pack_dst dst
end

module DMD_DMZ = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src = unpack_dmd

  let pack_dst = pack_dmz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_d2z src in
    pack_dst dst
end

(* lift both precision and element type *)

module DAS_DAZ = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src = unpack_das

  let pack_dst = pack_daz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2z src in
    pack_dst dst
end

module DMS_DMZ = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src = unpack_dms

  let pack_dst = pack_dmz

  let lift src =
    let src = unpack_src src in
    let dst = M.cast_s2z src in
    pack_dst dst
end

module DAC_DAD = struct
  module M = Owl_dense_ndarray.Generic

  let unpack_src0 = unpack_dac

  let unpack_src1 = unpack_dad

  let pack_dst = pack_daz

  let lift0 src =
    let src = unpack_src0 src in
    let dst = M.cast_c2z src in
    pack_dst dst


  let lift1 src =
    let src = unpack_src1 src in
    let dst = M.cast_d2z src in
    pack_dst dst
end

module DMC_DMD = struct
  module M = Owl_dense_matrix.Generic

  let unpack_src0 = unpack_dmc

  let unpack_src1 = unpack_dmd

  let pack_dst = pack_dmz

  let lift0 src =
    let src = unpack_src0 src in
    let dst = M.cast_c2z src in
    pack_dst dst


  let lift1 src =
    let src = unpack_src1 src in
    let dst = M.cast_d2z src in
    pack_dst dst
end

(* ends here *)