gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Changed dev wheel date to UTC time to match CRON schedule (#587)

Browse Source
This commit is contained in:
Philippe Tillet
2022-07-18 14:54:13 -07:00
committed by GitHub
parent 5b04331dd2
commit 86cab58d89
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/wheels.yml vendored
View File

@@ -18,7 +18,7 @@ jobs:
- name: Patch setup.py - name: Patch setup.py
run: | run: |
#sed -i 's/name\=\"triton\"/name="triton-nightly"/g' python/setup.py #sed -i 's/name\=\"triton\"/name="triton-nightly"/g' python/setup.py
export LATEST_DATE=$(git show -s --format=%ci `git rev-parse HEAD` | cut -d ' ' -f 1 | sed 's/-//g') export LATEST_DATE=$(TZ=UTC0 git show --quiet --date='format-local:%Y%m%d' --format="%cd")
sed -i -r "s/version\=\"(.*)\"/version=\"\1-dev"$LATEST_DATE"\"/g" python/setup.py sed -i -r "s/version\=\"(.*)\"/version=\"\1-dev"$LATEST_DATE"\"/g" python/setup.py
echo "" >> python/setup.cfg echo "" >> python/setup.cfg
echo "[build_ext]" >> python/setup.cfg echo "[build_ext]" >> python/setup.cfg

2
python/tutorials/06-fused-attention.py
View File

@@ -1,7 +1,7 @@
""" """
Fused Attention Fused Attention
=============== ===============
This is a Triton implementation of the Flash Attention algorithm This is a Triton implementation of the Flash Attention algorithm
(see: Dao et al., https://arxiv.org/pdf/2205.14135v2.pdf; Rabe and Staats https://arxiv.org/pdf/2112.05682v2.pdf) (see: Dao et al., https://arxiv.org/pdf/2205.14135v2.pdf; Rabe and Staats https://arxiv.org/pdf/2112.05682v2.pdf)
""" """