builds for pytorch 1.6

Summary: Make CI builds with pytorch 1.6.

Reviewed By: gkioxari

Differential Revision: D22790325

fbshipit-source-id: 5b0e075f952de9df2de03b65c16fd69140ab6fdd
This commit is contained in:
Jeremy Reizenstein 2020-07-28 10:38:45 -07:00 committed by Facebook GitHub Bot
parent 9a5341bde3
commit 326f662878
3 changed files with 72 additions and 26 deletions

View File

@ -212,22 +212,22 @@ workflows:
pytorch_version: "1.4" pytorch_version: "1.4"
cu_version: "cu100" cu_version: "cu100"
- binary_linux_conda_cuda: - binary_linux_conda_cuda:
name: testrun_conda_cuda_py37_cu102_pyt151 name: testrun_conda_cuda_py37_cu102_pyt160
python_version: "3.7" python_version: "3.7"
pytorch_version: "1.5.1" pytorch_version: '1.6.0'
cu_version: "cu102" cu_version: "cu102"
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py36_cpu name: macos_wheel_py36_cpu
python_version: '3.6' python_version: '3.6'
pytorch_version: '1.5' pytorch_version: '1.6.0'
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py37_cpu name: macos_wheel_py37_cpu
python_version: '3.7' python_version: '3.7'
pytorch_version: '1.5' pytorch_version: '1.6.0'
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py38_cpu name: macos_wheel_py38_cpu
python_version: '3.8' python_version: '3.8'
pytorch_version: '1.5' pytorch_version: '1.6.0'

View File

@ -250,6 +250,21 @@ workflows:
name: linux_conda_py36_cu102_pyt151 name: linux_conda_py36_cu102_pyt151
python_version: '3.6' python_version: '3.6'
pytorch_version: 1.5.1 pytorch_version: 1.5.1
- binary_linux_conda:
cu_version: cu92
name: linux_conda_py36_cu92_pyt160
python_version: '3.6'
pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu101
name: linux_conda_py36_cu101_pyt160
python_version: '3.6'
pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu102
name: linux_conda_py36_cu102_pyt160
python_version: '3.6'
pytorch_version: 1.6.0
- binary_linux_conda: - binary_linux_conda:
cu_version: cu92 cu_version: cu92
name: linux_conda_py37_cu92_pyt14 name: linux_conda_py37_cu92_pyt14
@ -295,6 +310,21 @@ workflows:
name: linux_conda_py37_cu102_pyt151 name: linux_conda_py37_cu102_pyt151
python_version: '3.7' python_version: '3.7'
pytorch_version: 1.5.1 pytorch_version: 1.5.1
- binary_linux_conda:
cu_version: cu92
name: linux_conda_py37_cu92_pyt160
python_version: '3.7'
pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu101
name: linux_conda_py37_cu101_pyt160
python_version: '3.7'
pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu102
name: linux_conda_py37_cu102_pyt160
python_version: '3.7'
pytorch_version: 1.6.0
- binary_linux_conda: - binary_linux_conda:
cu_version: cu92 cu_version: cu92
name: linux_conda_py38_cu92_pyt14 name: linux_conda_py38_cu92_pyt14
@ -340,43 +370,58 @@ workflows:
name: linux_conda_py38_cu102_pyt151 name: linux_conda_py38_cu102_pyt151
python_version: '3.8' python_version: '3.8'
pytorch_version: 1.5.1 pytorch_version: 1.5.1
- binary_linux_wheel: - binary_linux_conda:
cu_version: cpu cu_version: cu92
name: linux_wheel_py36_cpu_pyt15 name: linux_conda_py38_cu92_pyt160
python_version: '3.6'
pytorch_version: '1.5'
- binary_linux_wheel:
cu_version: cpu
name: linux_wheel_py37_cpu_pyt15
python_version: '3.7'
pytorch_version: '1.5'
- binary_linux_wheel:
cu_version: cpu
name: linux_wheel_py38_cpu_pyt15
python_version: '3.8' python_version: '3.8'
pytorch_version: '1.5' pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu101
name: linux_conda_py38_cu101_pyt160
python_version: '3.8'
pytorch_version: 1.6.0
- binary_linux_conda:
cu_version: cu102
name: linux_conda_py38_cu102_pyt160
python_version: '3.8'
pytorch_version: 1.6.0
- binary_linux_wheel:
cu_version: cpu
name: linux_wheel_py36_cpu_pyt160
python_version: '3.6'
pytorch_version: 1.6.0
- binary_linux_wheel:
cu_version: cpu
name: linux_wheel_py37_cpu_pyt160
python_version: '3.7'
pytorch_version: 1.6.0
- binary_linux_wheel:
cu_version: cpu
name: linux_wheel_py38_cpu_pyt160
python_version: '3.8'
pytorch_version: 1.6.0
- binary_linux_conda_cuda: - binary_linux_conda_cuda:
name: testrun_conda_cuda_py37_cu100_pyt14 name: testrun_conda_cuda_py37_cu100_pyt14
python_version: "3.7" python_version: "3.7"
pytorch_version: "1.4" pytorch_version: "1.4"
cu_version: "cu100" cu_version: "cu100"
- binary_linux_conda_cuda: - binary_linux_conda_cuda:
name: testrun_conda_cuda_py37_cu102_pyt151 name: testrun_conda_cuda_py37_cu102_pyt160
python_version: "3.7" python_version: "3.7"
pytorch_version: "1.5.1" pytorch_version: '1.6.0'
cu_version: "cu102" cu_version: "cu102"
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py36_cpu name: macos_wheel_py36_cpu
python_version: '3.6' python_version: '3.6'
pytorch_version: '1.5' pytorch_version: '1.6.0'
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py37_cpu name: macos_wheel_py37_cpu
python_version: '3.7' python_version: '3.7'
pytorch_version: '1.5' pytorch_version: '1.6.0'
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: macos_wheel_py38_cpu name: macos_wheel_py38_cpu
python_version: '3.8' python_version: '3.8'
pytorch_version: '1.5' pytorch_version: '1.6.0'

View File

@ -17,6 +17,7 @@ CONDA_CUDA_VERSIONS = {
"1.4": ["cu92", "cu100", "cu101"], "1.4": ["cu92", "cu100", "cu101"],
"1.5.0": ["cu92", "cu101", "cu102"], "1.5.0": ["cu92", "cu101", "cu102"],
"1.5.1": ["cu92", "cu101", "cu102"], "1.5.1": ["cu92", "cu101", "cu102"],
"1.6.0": ["cu92", "cu101", "cu102"],
} }
@ -24,7 +25,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
w = [] w = []
for btype in ["conda"]: for btype in ["conda"]:
for python_version in ["3.6", "3.7", "3.8"]: for python_version in ["3.6", "3.7", "3.8"]:
for pytorch_version in ["1.4", "1.5.0", "1.5.1"]: for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0"]:
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]: for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
w += workflow_pair( w += workflow_pair(
btype=btype, btype=btype,
@ -41,7 +42,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
w += workflow_pair( w += workflow_pair(
btype=btype, btype=btype,
python_version=python_version, python_version=python_version,
pytorch_version="1.5", pytorch_version="1.6.0",
cu_version=cu_version, cu_version=cu_version,
prefix=prefix, prefix=prefix,
upload=upload, upload=upload,