Intro-ops/requirements.txt

15 lines
208 B
Plaintext

# Python dependencies for operator_runtime_training
# Runtime and benchmarking (required)
torch>=2.0
# TileLang backend (optional)
tilelang
# Build tools
cmake>=3.22
ninja
# Testing framework
pytest>=7.0