search

pytorch-labs / tritonbench

Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
BSD 3-Clause "New" or "Revised" License
21 stars 3 forks source link

Add installation scripts. #2

Closed xuzhao9 closed 1 month ago

xuzhao9 commented 1 month ago

Supports running the following kernel repositories:

Test plan:

$ python install.py --all
INFO:__main__:[tritonbench] installing liger-kernels...
DEPRECATION: Loading egg at /home/xz/miniconda3/lib/python3.11/site-packages/flashattn_hopper-3.0.0b1-py3.11-linux-x86_64.egg is deprecated. pip 23.3 will enforce this behaviour change. A possible replacement is to use pip for package installation..
Requirement already satisfied: liger-kernel in /home/xz/miniconda3/lib/python3.11/site-packages (0.3.1)
INFO:__main__:[tritonbench] installation complete!