Source file owl_ext_lifts.ml

# 1 "src/owl/ext/owl_ext_lifts.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * Copyright (c) 2016-2018 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 *)