update

mjun0812 · Oct 19, 2024 · dc5f7a0 · dc5f7a0
1 parent 0ecd9c0
commit dc5f7a0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 15 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -80,20 +80,19 @@ jobs:
         run: |
           pip install -U pip
           pip install wheel setuptools packaging
-          export TORCH_CUDA_VERSION=$(python -c " \
-            from os import environ as env; \
-            support_cuda_versions = { \
-              '2.0': [117, 118], \
-              '2.1': [118, 121], \
-              '2.2': [118, 121], \
-              '2.3': [118, 121], \
-              '2.4': [118, 121, 124], \
-              '2.5': [118, 121, 124], \
-            }; \
-            target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \
-            cuda_version = int(env['MATRIX_CUDA_VERSION']); \
-            closest_version = min(target_cuda_versions, key=lambda x: abs(x - cuda_version)); \
-            print(closest_version) \
+          export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
+              support_cuda_versions = { \
+                  '2.0': [117, 118], \
+                  '2.1': [118, 121], \
+                  '2.2': [118, 121], \
+                  '2.3': [118, 121], \
+                  '2.4': [118, 121, 124], \
+                  '2.5': [118, 121, 124], \
+              }; \
+              target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \
+              cuda_version = int(env['MATRIX_CUDA_VERSION']); \
+              closest_version = min(target_cuda_versions, key=lambda x: abs(x - cuda_version)); \
+              print(closest_version) \
           ")
           pip install --no-cache-dir torch==${{ matrix.torch }} --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA_VERSION}
           nvcc --version

diff --git a/torch_cpp/__init__.py b/torch_cpp/__init__.py
@@ -1,3 +1,3 @@
 from .src import *  # noqa: F401, F403
 
-__version__ = "1.1.7"
+__version__ = "1.1.8"