From de8ecb556cb4dcc73926a654019fb24cacf5ff7a Mon Sep 17 00:00:00 2001 From: zhouzaida Date: Wed, 10 May 2023 20:00:10 +0800 Subject: [PATCH 1/2] [FIx] Build CUDA ops in CI --- .github/workflows/merge_stage_test.yml | 9 +++++++++ .github/workflows/pr_stage_test.yml | 3 +++ 2 files changed, 12 insertions(+) diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml index 966ff8d4dc..d27732367b 100644 --- a/.github/workflows/merge_stage_test.yml +++ b/.github/workflows/merge_stage_test.yml @@ -175,6 +175,9 @@ jobs: runs-on: ubuntu-22.04 container: image: pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel + env: + FORCE_CUDA: 1 + MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61 strategy: matrix: python-version: [3.7] @@ -209,6 +212,9 @@ jobs: runs-on: ubuntu-22.04 container: image: pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel + env: + FORCE_CUDA: 1 + MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61 strategy: matrix: python-version: [3.7] @@ -243,6 +249,9 @@ jobs: runs-on: ubuntu-22.04 container: image: pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel + env: + FORCE_CUDA: 1 + MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61 strategy: matrix: python-version: [3.7] diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml index dde09127c5..fe2d9bc52e 100644 --- a/.github/workflows/pr_stage_test.yml +++ b/.github/workflows/pr_stage_test.yml @@ -23,6 +23,9 @@ jobs: runs-on: ubuntu-22.04 container: image: pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel + env: + FORCE_CUDA: 1 + MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61 strategy: matrix: python-version: [3.7] From 8a633389dc13a523bbc4011450d216c993dc2e96 Mon Sep 17 00:00:00 2001 From: zhouzaida Date: Wed, 10 May 2023 20:01:24 +0800 Subject: [PATCH 2/2] update --- .github/workflows/pr_stage_test.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml index fe2d9bc52e..d790384f14 100644 --- a/.github/workflows/pr_stage_test.yml +++ b/.github/workflows/pr_stage_test.yml @@ -68,6 +68,9 @@ jobs: runs-on: ubuntu-22.04 container: image: pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel + env: + FORCE_CUDA: 1 + MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61 strategy: matrix: python-version: [3.7]