Experimental

The kernels in this folder are experimental, and for now not tied to the rest of the repo.

Requirements

• Either: -- Checkout the Triton v2.0 branch git clone [email protected]:openai/triton.git; cd triton; git checkout v2.0

-- Install it in a dedicated environement. Note that you will need pytorch, and a matching GCC/G++/CUDA cd triton/python; pip install -e .

• or install the Triton 2 dev wheel: -- pip install -r requirements.txt

• (Optional) install the kernels present here by running pip install -e . in this folder

• (Optional) Check that the installation is successful by running pytest tests in this folder

If things go south, grab a backtrace as follows

gdb -batch -ex run -ex bt --args python -m pytest [..]