klr-lang 0.0.5

Intermediate langauge for tensor compilers

Python bindings for KLR

Usage

Note

There are 3 files in our nki Python package. They are all copied directly from the Neuron SDK repo:

• nki/init.py
• nki/isa/init.py
• nki/language/init.py

We do this to remove a dependency on the neuronxcc Python package that does not build on OSX.

TODO

1. Automatically sync nki API directories with the neuronxcc package. The later does not install on OSX, and we generally don't want to require it since it's large and unnecessary.

2. Automatically sync nki examples with the github examples

Notes

I keep getting pycache directories created in the source tree, even with

export PYTHONDONTWRITEBYTECODE=1

This is annoying.