Gpu

GPU computation primitives. This module is imported automatically by the ``@gpu`` decorator, or can be brought in explicitly with ``open Gpu``.

Table of Contents

Types

(type)

type GpuConfig

def gpu_map (f: (x : a) → b) (t: Tensor) : Tensor

def gpu_imap (f: (i : Int) → b) (t: Tensor) : Tensor

def gpu_reduce (f: (acc : a) → (x : a) → a) (initial: a) (t: Tensor) : a

def gpu_filter (f: (x : a) → Bool) (t: Tensor) : Tensor

def gpu_dot (a: Tensor) (b: Tensor) : Float

def run_gpu (kernel: (x : Tensor) → a) (config: GpuConfig) (input: Tensor) : a