This document is relevant for: Trn2, Trn3
nki#
The nki module provides the top-level entry points for compiling and running NKI kernels.
Use the jit() decorator to compile a kernel for NeuronDevices, or simulate() to
run a kernel in the CPU simulator for debugging.
This decorator compiles a top-level NKI function to run on NeuronDevices. |
|
Create a CPU-simulated version of an NKI kernel. |
This document is relevant for: Trn2, Trn3