search

kokkos / kokkos.github.io

Source code for kokkos.org pages
https://kokkos.org
1 stars 12 forks source link

Kokkos Kernels - ML/IR #81

Open crtrott opened 10 months ago

crtrott commented 10 months ago

Authors: @brian-kelley @kliegeois

brian-kelley commented 9 months ago

The MLIR (multi-level intermediate representation) in LLVM can represent operations ranging from high-level (e.g. matrix multiplication) to low-level (e.g. a bitwise OR). It also includes automatic transformations for lowering high-level operations to equivalent sequences of low-level operations. We present an MLIR-based compiler pipeline that can compile PyTorch models and general sparse tensor expressions into parallel Kokkos C++ source code. We also discuss ongoing work involving Kokkos and MLIR.

brian-kelley commented 9 months ago

Kokkos_MLIR.pdf