fix release cu118 pkg (#10325)

Oneflow-Inc · Sep 1, 2023 · f31e60c · f31e60c
1 parent dabb27a
commit f31e60c
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 4 deletions.
diff --git a/.github/workflows/canary.yml b/.github/workflows/canary.yml
@@ -72,8 +72,8 @@ jobs:
           clean-ccache: true
           compute-platform: ${{ env.COMPUTE_PLATFORM }}
           python-versions: |
-            3.7
             3.8
+            3.10
       - name: Upload wheelhouse
         uses: ./.github/actions/upload_oss
         with:

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -48,7 +48,6 @@ jobs:
           entries: |
             cu118
             cu117
-            cu116
             cpu
       - name: Get current date
         id: date
@@ -80,7 +79,32 @@ jobs:
       - uses: actions/checkout@v2
       - uses: Oneflow-Inc/get-oneflow@add-nightly-date
         name: Build ${{ matrix.entry }}
-        if: ${{ matrix.entry !='cpu' }}
+        if: ${{ matrix.entry =='cu118' }}
+        with:
+          cmake-init-cache: ${{ env.ONEFLOW_SRC }}/cmake/caches/ci/release/${{ matrix.entry }}.cmake
+          build-script: ${{ env.ONEFLOW_SRC }}/ci/manylinux/build-gcc9.sh
+          oneflow-src: ${{ env.ONEFLOW_SRC }}
+          oneflow-build-env: manylinux
+          wheelhouse-dir: ${{ env.WHEELHOUSE_DIR }}
+          clear-wheelhouse-dir: true
+          self-hosted: true
+          compute-platform: ${{ matrix.entry }}
+          manylinux-cache-dir: ${{ env.MANYLINUX_CACHE_DIR }}
+          docker-run-use-system-http-proxy: false
+          docker-run-use-lld: false
+          retry-failed-build: true
+          clean-ccache: true
+          nightly: ${{ github.event_name == 'schedule' || github.ref == 'refs/heads/release/add_nightly_date_index'}}
+          nightly-date: ${{ needs.generate-build-matrix.outputs.formatted_date }}
+          use-nvidia-wheels: ${{ matrix.entry !='cu112' }}
+          python-versions: |
+            3.7
+            3.8
+            3.9
+            3.10
+      - uses: Oneflow-Inc/get-oneflow@add-nightly-date
+        name: Build ${{ matrix.entry }}
+        if: ${{ matrix.entry !='cpu' && matrix.entry !='cu118' }}
         with:
           cmake-init-cache: ${{ env.ONEFLOW_SRC }}/cmake/caches/ci/release/cuda.cmake
           build-script: ${{ env.ONEFLOW_SRC }}/ci/manylinux/build-gcc9.sh

diff --git a/README.md b/README.md
@@ -1,6 +1,7 @@
 # OneFlow
 
 OneFlow is a deep learning framework designed to be **user-friendly, scalable and efficient**. With OneFlow, it is easy to:
+
 - program a model with [**PyTorch-like API**](https://oneflow.readthedocs.io/en/master/)
 - scale a model to n-dimensional-parallel execution with the [**Global Tensor**](https://docs.oneflow.org/en/master/cookies/global_tensor.html)
 - accelerate/deploy a model with the [**Graph Compiler**](https://oneflow.readthedocs.io/en/master/graph.html).
@@ -77,7 +78,6 @@ OneFlow is a deep learning framework designed to be **user-friendly, scalable an
     | Platform |CUDA Driver Version| Supported GPUs |
     |---|---|---|
     | cu117 | >= 450.80.02 | GTX 10xx, RTX 20xx, A100, RTX 30xx |
-    | cu102 | >= 440.33 | GTX 10xx, RTX 20xx |
     | cpu | N/A | N/A |
 
 - If you are in China, you could run this to have pip download packages from domestic mirror of pypi:

diff --git a/cmake/caches/ci/release/cu118.cmake b/cmake/caches/ci/release/cu118.cmake
@@ -0,0 +1,17 @@
+set(BUILD_CUDA YES CACHE BOOL "")
+set(BUILD_GIT_VERSION YES CACHE BOOL "")
+set(BUILD_TESTING OFF CACHE BOOL "")
+set(BUILD_RDMA YES CACHE BOOL "")
+set(TREAT_WARNINGS_AS_ERRORS YES CACHE BOOL "")
+set(THIRD_PARTY_MIRROR aliyun CACHE STRING "")
+set(PIP_INDEX_MIRROR "https://pypi.tuna.tsinghua.edu.cn/simple" CACHE STRING "")
+set(CMAKE_BUILD_TYPE Release CACHE STRING "")
+set(CMAKE_GENERATOR Ninja CACHE STRING "")
+set(CMAKE_CUDA_ARCHITECTURES "70-real;80-real;86-real;89-real;90-real" CACHE STRING "")
+set(CUDNN_STATIC OFF CACHE BOOL "")
+set(WITH_MLIR ON CACHE BOOL "")
+set(BUILD_CPP_API OFF CACHE BOOL "")
+set(CUDA_NVCC_THREADS_NUMBER 2 CACHE STRING "")
+set(CMAKE_C_COMPILER_LAUNCHER ccache CACHE STRING "")
+set(CMAKE_CXX_COMPILER_LAUNCHER ccache CACHE STRING "")
+set(CMAKE_CUDA_COMPILER_LAUNCHER ccache CACHE STRING "")