fix

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

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -82,25 +82,25 @@ jobs:
           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],
+            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 = env['MATRIX_CUDA_VERSION']; \
-            closest_version = min(target_cuda_versions, key=lambda x: abs(x - int(cuda_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
           python --version
           python -c "import torch; print('PyTorch:', torch.__version__)"
           python -c "import torch; print('CUDA:', torch.version.cuda)"
-          python -c "from torch.utils import cpp_extension; print (cpp_extension.CUDA_HOME)"
+          python -c "from torch.utils import cpp_extension; print(cpp_extension.CUDA_HOME)"
 
       - name: Build
         run: |

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.6"
+__version__ = "1.1.7"