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 "")