Triton 1.0.0 required for sparse attention but does not exist #2408
Unanswered
maxmaier59
asked this question in
Q&A
Replies: 1 comment 1 reply
-
Requires clang and llvm 11, specifically. #!/usr/bin/env bash
# Downloads and unzips Triton source code
wget -O triton-1.0.zip https://github.com/openai/triton/archive/refs/tags/v1.0.zip
unzip triton-1.0.zip
rm triton-1.0.zip
mv triton-1.0 triton
# Patches triton so it compiles
cd triton
{ head -n 9 include/triton/tools/graph.h; echo "using std::size_t;"; tail -n +10 include/triton/tools/graph.h; } > include/triton/tools/graph.new.h
mv include/triton/tools/graph.new.h include/triton/tools/graph.h
# Installs triton
cd python
pip3 install -e . |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
For sparse attention Triton 1.0.0 is required:
Only this version works with DeepSpeed
DeepSpeed C++/CUDA extension op report
NOTE: Ops not installed will be just-in-time (JIT) compiled at
runtime if needed. Op compatibility means that your system
meet the required dependencies to JIT install the op.
JIT compiled ops requires ninja
ninja .................. [OKAY]
op name ................ installed .. compatible
cpu_adam ............... [NO] ....... [OKAY]
cpu_adagrad ............ [NO] ....... [OKAY]
fused_adam ............. [NO] ....... [OKAY]
fused_lamb ............. [NO] ....... [OKAY]
[WARNING] please install triton==1.0.0 if you want to use sparse attention
sparse_attn ............ [NO] ....... [NO]
transformer ............ [NO] ....... [OKAY]
stochastic_transformer . [NO] ....... [OKAY]
async_io ............... [NO] ....... [OKAY]
utils .................. [NO] ....... [OKAY]
quantizer .............. [NO] ....... [OKAY]
transformer_inference .. [NO] ....... [OKAY]
However, Trition 1.0.0 does not exist:
$ pip install triton==1.0.0
ERROR: Could not find a version that satisfies the requirement triton==1.0.0 (from versions: 0.1, 0.1.1, 0.1.2, 0.1.3, 0.2.0, 0.2.1, 0.2.2, 0.2.3, 0.3.0)
ERROR: No matching distribution found for triton==1.0.0
What is the solution???
Beta Was this translation helpful? Give feedback.
All reactions