triton — Easy Playground

OpenAI Triton: write custom GPU kernels in Python

Getting started with tritonRun locally

Install

pip install triton

Python CodeRun locally

# Install: pip install triton
import triton

# Basic triton usage
print(f"Using triton")
# See documentation for detailed examples

# Install: pip install triton
import triton

# Basic triton usage
print(f"Using triton")
# See documentation for detailed examples

Expected Output

# Expected output shown below
# (Run locally with: triton)

triton is a third-party package. OpenAI Triton: write custom GPU kernels in Python. Install with: pip install triton

Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?