From 23cf3fb8b36b1a82446c81227ca59f1f0772770e Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Mon, 26 Apr 2021 17:58:01 -0500 Subject: [PATCH 1/3] add rocm binary use the term cu_version update pkg_helpers add docker wheel option update config yml add force_cuda option --- .circleci/regenerate.py | 6 ++++-- packaging/pkg_helpers.bash | 4 ++++ 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index f1720979fb..61120a1f85 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -21,7 +21,7 @@ PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"] -CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111"], +CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111","rocm4.1"], "windows": ["cpu", "cu102", "cu111"], "macos": ["cpu"]} @@ -124,8 +124,10 @@ def generate_base_workflow(base_workflow_name, python_version, cu_version, filte d['requires'] = ['download_third_parties_nix'] if btype == 'conda': d['conda_docker_image'] = f'pytorch/conda-builder:{cu_version.replace("cu1","cuda1")}' - elif cu_version != 'cpu': + elif cu_version.startswith('cu'): d['wheel_docker_image'] = f'pytorch/manylinux-{cu_version.replace("cu1","cuda1")}' + elif cu_version.startswith('rocm'): + d["wheel_docker_image"] = f"pytorch/manylinux-rocm:{cu_version[len('rocm'):]}" if filter_branch: d["filters"] = gen_filter_branch_tree(filter_branch) diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash index 173aa4ca07..1626938863 100644 --- a/packaging/pkg_helpers.bash +++ b/packaging/pkg_helpers.bash @@ -107,6 +107,10 @@ setup_cuda() { export FORCE_CUDA=1 export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0" ;; + rocm*) + export FORCE_CUDA=1 + export USE_ROCM=1 + ;; cpu) ;; *) From df1e434a91add754302d65e87d7166fdc54a23ff Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Thu, 27 May 2021 12:53:53 -0700 Subject: [PATCH 2/3] update config.yml --- .circleci/config.yml | 344 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 344 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index 32c7effa9c..d90b1f9ef1 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -650,6 +650,13 @@ workflows: requires: - download_third_parties_nix wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: rocm4.1 + name: binary_linux_wheel_py3.6_rocm4.1 + python_version: '3.6' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 - binary_linux_wheel: cuda_version: cpu name: binary_linux_wheel_py3.7_cpu @@ -670,6 +677,13 @@ workflows: requires: - download_third_parties_nix wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: rocm4.1 + name: binary_linux_wheel_py3.7_rocm4.1 + python_version: '3.7' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 - binary_linux_wheel: cuda_version: cpu name: binary_linux_wheel_py3.8_cpu @@ -690,6 +704,13 @@ workflows: requires: - download_third_parties_nix wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: rocm4.1 + name: binary_linux_wheel_py3.8_rocm4.1 + python_version: '3.8' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 - binary_linux_wheel: cuda_version: cpu name: binary_linux_wheel_py3.9_cpu @@ -710,6 +731,13 @@ workflows: requires: - download_third_parties_nix wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: rocm4.1 + name: binary_linux_wheel_py3.9_rocm4.1 + python_version: '3.9' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 - binary_macos_wheel: cuda_version: cpu name: binary_macos_wheel_py3.6_cpu @@ -811,6 +839,13 @@ workflows: python_version: '3.6' requires: - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + name: binary_linux_conda_py3.6_rocm4.1 + python_version: '3.6' + requires: + - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -832,6 +867,13 @@ workflows: python_version: '3.7' requires: - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + name: binary_linux_conda_py3.7_rocm4.1 + python_version: '3.7' + requires: + - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -853,6 +895,13 @@ workflows: python_version: '3.8' requires: - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + name: binary_linux_conda_py3.8_rocm4.1 + python_version: '3.8' + requires: + - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -874,6 +923,13 @@ workflows: python_version: '3.9' requires: - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + name: binary_linux_conda_py3.9_rocm4.1 + python_version: '3.9' + requires: + - download_third_parties_nix - binary_macos_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -1217,6 +1273,42 @@ workflows: python_version: '3.6' requires: - nightly_binary_linux_wheel_py3.6_cu111_upload + - binary_linux_wheel: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.6_rocm4.1 + python_version: '3.6' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload + requires: + - nightly_binary_linux_wheel_py3.6_rocm4.1 + - smoke_test_linux_pip: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip + python_version: '3.6' + requires: + - nightly_binary_linux_wheel_py3.6_rocm4.1_upload - binary_linux_wheel: cuda_version: cpu filters: @@ -1324,6 +1416,42 @@ workflows: python_version: '3.7' requires: - nightly_binary_linux_wheel_py3.7_cu111_upload + - binary_linux_wheel: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.7_rocm4.1 + python_version: '3.7' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload + requires: + - nightly_binary_linux_wheel_py3.7_rocm4.1 + - smoke_test_linux_pip: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip + python_version: '3.7' + requires: + - nightly_binary_linux_wheel_py3.7_rocm4.1_upload - binary_linux_wheel: cuda_version: cpu filters: @@ -1431,6 +1559,42 @@ workflows: python_version: '3.8' requires: - nightly_binary_linux_wheel_py3.8_cu111_upload + - binary_linux_wheel: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.8_rocm4.1 + python_version: '3.8' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload + requires: + - nightly_binary_linux_wheel_py3.8_rocm4.1 + - smoke_test_linux_pip: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip + python_version: '3.8' + requires: + - nightly_binary_linux_wheel_py3.8_rocm4.1_upload - binary_linux_wheel: cuda_version: cpu filters: @@ -1538,6 +1702,42 @@ workflows: python_version: '3.9' requires: - nightly_binary_linux_wheel_py3.9_cu111_upload + - binary_linux_wheel: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.9_rocm4.1 + python_version: '3.9' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload + requires: + - nightly_binary_linux_wheel_py3.9_rocm4.1 + - smoke_test_linux_pip: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip + python_version: '3.9' + requires: + - nightly_binary_linux_wheel_py3.9_rocm4.1_upload - binary_macos_wheel: cuda_version: cpu filters: @@ -2142,6 +2342,42 @@ workflows: python_version: '3.6' requires: - nightly_binary_linux_conda_py3.6_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_rocm4.1 + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_rocm4.1_upload + requires: + - nightly_binary_linux_conda_py3.6_rocm4.1 + - smoke_test_linux_conda: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_rocm4.1_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_linux_conda_py3.6_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2250,6 +2486,42 @@ workflows: python_version: '3.7' requires: - nightly_binary_linux_conda_py3.7_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_rocm4.1 + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_rocm4.1_upload + requires: + - nightly_binary_linux_conda_py3.7_rocm4.1 + - smoke_test_linux_conda: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_rocm4.1_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_linux_conda_py3.7_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2358,6 +2630,42 @@ workflows: python_version: '3.8' requires: - nightly_binary_linux_conda_py3.8_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_rocm4.1 + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_rocm4.1_upload + requires: + - nightly_binary_linux_conda_py3.8_rocm4.1 + - smoke_test_linux_conda: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_rocm4.1_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_linux_conda_py3.8_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2466,6 +2774,42 @@ workflows: python_version: '3.9' requires: - nightly_binary_linux_conda_py3.9_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:rocm4.1 + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_rocm4.1 + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_rocm4.1_upload + requires: + - nightly_binary_linux_conda_py3.9_rocm4.1 + - smoke_test_linux_conda: + cuda_version: rocm4.1 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_rocm4.1_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_linux_conda_py3.9_rocm4.1_upload - binary_macos_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu From 8f3231cc71eb6b40faad8603e01c2f71cea2d438 Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Thu, 27 May 2021 16:31:36 -0700 Subject: [PATCH 3/3] remove conda builds --- .circleci/config.yml | 172 ---------------------------------------- .circleci/regenerate.py | 5 +- 2 files changed, 4 insertions(+), 173 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index d90b1f9ef1..cb796d03eb 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -839,13 +839,6 @@ workflows: python_version: '3.6' requires: - download_third_parties_nix - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - name: binary_linux_conda_py3.6_rocm4.1 - python_version: '3.6' - requires: - - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -867,13 +860,6 @@ workflows: python_version: '3.7' requires: - download_third_parties_nix - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - name: binary_linux_conda_py3.7_rocm4.1 - python_version: '3.7' - requires: - - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -895,13 +881,6 @@ workflows: python_version: '3.8' requires: - download_third_parties_nix - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - name: binary_linux_conda_py3.8_rocm4.1 - python_version: '3.8' - requires: - - download_third_parties_nix - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -923,13 +902,6 @@ workflows: python_version: '3.9' requires: - download_third_parties_nix - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - name: binary_linux_conda_py3.9_rocm4.1 - python_version: '3.9' - requires: - - download_third_parties_nix - binary_macos_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2342,42 +2314,6 @@ workflows: python_version: '3.6' requires: - nightly_binary_linux_conda_py3.6_cu111_upload - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6_rocm4.1 - python_version: '3.6' - requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6_rocm4.1_upload - requires: - - nightly_binary_linux_conda_py3.6_rocm4.1 - - smoke_test_linux_conda: - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6_rocm4.1_smoke_test_conda - python_version: '3.6' - requires: - - nightly_binary_linux_conda_py3.6_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2486,42 +2422,6 @@ workflows: python_version: '3.7' requires: - nightly_binary_linux_conda_py3.7_cu111_upload - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7_rocm4.1 - python_version: '3.7' - requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7_rocm4.1_upload - requires: - - nightly_binary_linux_conda_py3.7_rocm4.1 - - smoke_test_linux_conda: - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7_rocm4.1_smoke_test_conda - python_version: '3.7' - requires: - - nightly_binary_linux_conda_py3.7_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2630,42 +2530,6 @@ workflows: python_version: '3.8' requires: - nightly_binary_linux_conda_py3.8_cu111_upload - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8_rocm4.1 - python_version: '3.8' - requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8_rocm4.1_upload - requires: - - nightly_binary_linux_conda_py3.8_rocm4.1 - - smoke_test_linux_conda: - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8_rocm4.1_smoke_test_conda - python_version: '3.8' - requires: - - nightly_binary_linux_conda_py3.8_rocm4.1_upload - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu @@ -2774,42 +2638,6 @@ workflows: python_version: '3.9' requires: - nightly_binary_linux_conda_py3.9_cu111_upload - - binary_linux_conda: - conda_docker_image: pytorch/conda-builder:rocm4.1 - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9_rocm4.1 - python_version: '3.9' - requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9_rocm4.1_upload - requires: - - nightly_binary_linux_conda_py3.9_rocm4.1 - - smoke_test_linux_conda: - cuda_version: rocm4.1 - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9_rocm4.1_smoke_test_conda - python_version: '3.9' - requires: - - nightly_binary_linux_conda_py3.9_rocm4.1_upload - binary_macos_conda: conda_docker_image: pytorch/conda-builder:cpu cuda_version: cpu diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 61120a1f85..0f64498754 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -36,7 +36,10 @@ def build_workflows(prefix='', upload=False, filter_branch=None, indentation=6): for os_type in ["linux", "macos", "windows"]: for python_version in PYTHON_VERSIONS: for cu_version in CU_VERSIONS_DICT[os_type]: - w += build_workflow_pair(btype, os_type, python_version, cu_version, filter_branch, prefix, upload) + if cu_version.startswith("rocm") and btype=="conda": + pass + else: + w += build_workflow_pair(btype, os_type, python_version, cu_version, filter_branch, prefix, upload) if not filter_branch: # Build on every pull request, but upload only on nightly and tags