Use torch 2.1.0.dev20230731

tridao · Aug 12, 2023 · 56a35d2 · 56a35d2
1 parent 157af69
commit 56a35d2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml
@@ -42,13 +42,13 @@ jobs:
       matrix:
           os: [ubuntu-20.04]
           python-version: ['3.7', '3.8', '3.9', '3.10']
-          torch-version: ['1.12.1', '1.13.1', '2.0.1', '2.1.0.dev20230613']
+          torch-version: ['1.12.1', '1.13.1', '2.0.1', '2.1.0.dev20230731']
           cuda-version: ['11.6.2', '11.7.1', '11.8.0', '12.1.0']
           exclude:
             # Pytorch >= 2.0 only supports Python >= 3.8
             - torch-version: '2.0.1'
               python-version: '3.7'
-            - torch-version: '2.1.0.dev20230613'
+            - torch-version: '2.1.0.dev20230731'
               python-version: '3.7'
             # Pytorch <= 2.0 only supports CUDA <= 11.8
             - torch-version: '1.12.1'
@@ -58,11 +58,11 @@ jobs:
             - torch-version: '2.0.1'
               cuda-version: '12.1.0'
             # Pytorch >= 2.1 only supports CUDA 12.1
-            - torch-version: '2.1.0.dev20230613'
+            - torch-version: '2.1.0.dev20230731'
               cuda-version: '11.6.2'
-            - torch-version: '2.1.0.dev20230613'
+            - torch-version: '2.1.0.dev20230731'
               cuda-version: '11.7.1'
-            - torch-version: '2.1.0.dev20230613'
+            - torch-version: '2.1.0.dev20230731'
               cuda-version: '11.8.0'
 
     steps:

diff --git a/flash_attn/__init__.py b/flash_attn/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.8.post8"
+__version__ = "2.0.8.post9"
 
 from flash_attn.flash_attn_interface import flash_attn_func
 from flash_attn.flash_attn_interface import flash_attn_kvpacked_func