From 86cab58d89909aebd8970ca80a3e85220dd033d9 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Mon, 18 Jul 2022 14:54:13 -0700 Subject: [PATCH] [CI] Changed dev wheel date to UTC time to match CRON schedule (#587) --- .github/workflows/wheels.yml | 2 +- python/tutorials/06-fused-attention.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index 1d8d450f2..db682f33f 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -18,7 +18,7 @@ jobs: - name: Patch setup.py run: | #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 echo "" >> python/setup.cfg echo "[build_ext]" >> python/setup.cfg diff --git a/python/tutorials/06-fused-attention.py b/python/tutorials/06-fused-attention.py index c19ee498a..fb0f4f958 100644 --- a/python/tutorials/06-fused-attention.py +++ b/python/tutorials/06-fused-attention.py @@ -1,7 +1,7 @@ """ 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) """