Module `Sarek.Transform`Source

Sourceval a_to_vect : Kirc_Ast.k_ext -> Kirc_Ast.k_ext

val a_to_return_vect : 
  Kirc_Ast.k_ext ->
  Kirc_Ast.k_ext ->
  Kirc_Ast.k_ext ->
  Kirc_Ast.k_ext

Sourceval param_list : int list ref

Sourceval add_to_param_list : int -> unit

Sourceval check_and_transform_to_map : Kirc_Ast.k_ext -> Kirc_Ast.k_ext

Sourceval arg_of_vec : ('a, 'b) Spoc.Vector.vector -> ('a, 'b) Spoc.Kernel.kernelArgs

Source

val launch_kernel_with_args : 
  Spoc__Kernel.kernel ->
  Spoc__Kernel.grid ->
  Spoc__Kernel.block ->
  ('a, 'b) Spoc__Kernel.kernelArgs array ->
  Spoc.Devices.device ->
  unit

Source

val compute_grid_block_1D : 
  Spoc.Devices.device ->
  ('a, 'b) Spoc.Vector.vector ->
  Spoc.Kernel.grid * Spoc.Kernel.block

Source

val propagate : 
  (Kirc_Ast.k_ext -> Kirc_Ast.k_ext) ->
  Kirc_Ast.k_ext ->
  Kirc_Ast.k_ext

Source

val _map : 
  ('a, 'b, 'c -> 'd, 'd, 'j) Kirc.sarek_kernel ->
  ?dev:Spoc.Devices.device ->
  ('c, 'h) Spoc.Vector.vector ->
  ('d, 'j) Spoc.Vector.vector

Source

val map2 : 
  ('a, 'b, 'c -> 'd -> 'e, 'e, 'm) Kirc.sarek_kernel ->
  ?dev:Spoc.Devices.device ->
  ('c, 'i) Spoc.Vector.vector ->
  ('d, 'k) Spoc.Vector.vector ->
  ('e, 'm) Spoc.Vector.vector

Source

val reduce : 
  ('a, 'b, 'c -> 'c -> 'd, 'e, 'f) Kirc.sarek_kernel ->
  ?dev:Spoc.Devices.device ->
  ('c, 'i) Spoc.Vector.vector ->
  'd

Sourceval (^>) : string -> string -> string

Source

val build_new_ker : 
  'a ->
  ('b, 'c, 'd) Kirc.kirc_kernel ->
  Kirc_Ast.k_ext ->
  'e ->
  'a * ('e, unit, unit) Kirc.kirc_kernel

Source

val map : 
  ('a, 'b, 'c -> 'f, 'f, 'g) Kirc.sarek_kernel ->
  ?dev:Spoc.Devices.device ->
  ('f, 'g) Spoc.Vector.vector ->
  ('f, 'g) Spoc.Vector.vector

Sourceexception Zip of string

Source

val zip : 
  ('a, 'b, 'c -> 'd -> 'j, 'j, 'k) Kirc.sarek_kernel ->
  ?dev:Spoc.Devices.device ->
  ('j, 'k) Spoc.Vector.vector ->
  ('j, 'k) Spoc.Vector.vector ->
  ('j, 'k) Spoc.Vector.vector

package sarek

Module Sarek.TransformSource

Module `Sarek.Transform`Source