Skip to main content

Python-to-PTX GPU Kernel Compiler with MLIR optimization

Project description

Oven Compiler

Python-to-PTX GPU compiler using MLIR.

Installation

pip install oven-compiler

Quick Start

Command Line

# Compile Python to PTX
oven-compiler --python kernel.py -o output.ptx

Python API

import oven_compiler

python_code = "def add(a, b): return a + b"
ptx_code = oven_compiler.compile_python_string_to_ptx(python_code)

Requirements

  • Python ≥ 3.12
  • CUDA-capable GPU (for execution)

License

MIT License

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

oven_compiler-0.1.7-cp312-cp312-manylinux2014_x86_64.whl (58.0 MB view details)

Uploaded CPython 3.12

File details

Details for the file oven_compiler-0.1.7-cp312-cp312-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for oven_compiler-0.1.7-cp312-cp312-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 402266db7da9ac3b39902ae547598efaf032c35627853cffccb8f5e4b18316d5
MD5 fd6723c01b804735dc161e4767ee4078
BLAKE2b-256 9923e98fd5e1bda4806d4b25c180439474f7ef8b56f539a180a5ff23796505ca

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page