oven-language 0.1.8

Python-to-MLIR compiler for GPU kernel development and mathematical computing

Oven Language

A Python-to-MLIR compiler for GPU kernel development.

Installation

pip install oven-language

Quick Start

import oven.language as ol

def vector_add(a_ptr: ol.ptr, b_ptr: ol.ptr, c_ptr: ol.ptr):
    bsize = ol.get_bdim_x()
    bid = ol.get_bid_x()
    tid = ol.get_tid_x()
    idx = (bid * bsize + tid) * 4
    x_value = ol.vload(a_ptr, idx)
    y_value = ol.vload(b_ptr, idx)
    z_value = x_value + y_value
    ol.vstore(z_value, c_ptr, idx)

Compile to MLIR:

oven kernel.py

Key Functions

• Thread/Block IDs: ol.get_tid_x(), ol.get_bid_x()
• Memory: ol.load(), ol.store(), ol.vload(), ol.vstore()
• Math: ol.exp(), ol.log(), ol.sin(), ol.sigmoid()
• GPU: ol.barrier(), ol.smem()

Links

• GitHub
• PyPI