builds for pytorch 1.6

Summary: Make CI builds with pytorch 1.6.

Reviewed By: gkioxari

Differential Revision: D22790325

fbshipit-source-id: 5b0e075f952de9df2de03b65c16fd69140ab6fdd

.circleci/config.in.yml

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

.circleci/config.yml

@@ -250,6 +250,21 @@ workflows:
           name: linux_conda_py36_cu102_pyt151
           python_version: '3.6'
           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:
           cu_version: cu92
           name: linux_conda_py37_cu92_pyt14
@@ -295,6 +310,21 @@ workflows:
           name: linux_conda_py37_cu102_pyt151
           python_version: '3.7'
           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:
           cu_version: cu92
           name: linux_conda_py38_cu92_pyt14
@@ -340,43 +370,58 @@ workflows:
           name: linux_conda_py38_cu102_pyt151
           python_version: '3.8'
           pytorch_version: 1.5.1
+      - binary_linux_conda:
+          cu_version: cu92
+          name: linux_conda_py38_cu92_pyt160
+          python_version: '3.8'
+          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_pyt15
+          name: linux_wheel_py36_cpu_pyt160
           python_version: '3.6'
-          pytorch_version: '1.5'
+          pytorch_version: 1.6.0
       - binary_linux_wheel:
           cu_version: cpu
-          name: linux_wheel_py37_cpu_pyt15
+          name: linux_wheel_py37_cpu_pyt160
           python_version: '3.7'
-          pytorch_version: '1.5'
+          pytorch_version: 1.6.0
       - binary_linux_wheel:
           cu_version: cpu
-          name: linux_wheel_py38_cpu_pyt15
+          name: linux_wheel_py38_cpu_pyt160
           python_version: '3.8'
-          pytorch_version: '1.5'
+          pytorch_version: 1.6.0
       - binary_linux_conda_cuda:
           name: testrun_conda_cuda_py37_cu100_pyt14
           python_version: "3.7"
           pytorch_version: "1.4"
           cu_version: "cu100"
       - binary_linux_conda_cuda:
-          name: testrun_conda_cuda_py37_cu102_pyt151
+          name: testrun_conda_cuda_py37_cu102_pyt160
           python_version: "3.7"
-          pytorch_version: "1.5.1"
+          pytorch_version: '1.6.0'
           cu_version: "cu102"
       - binary_macos_wheel:
           cu_version: cpu
           name: macos_wheel_py36_cpu
           python_version: '3.6'
-          pytorch_version: '1.5'
+          pytorch_version: '1.6.0'
       - binary_macos_wheel:
           cu_version: cpu
           name: macos_wheel_py37_cpu
           python_version: '3.7'
-          pytorch_version: '1.5'
+          pytorch_version: '1.6.0'
       - binary_macos_wheel:
           cu_version: cpu
           name: macos_wheel_py38_cpu
           python_version: '3.8'
-          pytorch_version: '1.5'
+          pytorch_version: '1.6.0'

.circleci/regenerate.py

@@ -17,6 +17,7 @@ CONDA_CUDA_VERSIONS = {
     "1.4": ["cu92", "cu100", "cu101"],
     "1.5.0": ["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 = []
     for btype in ["conda"]:
         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]:
                     w += workflow_pair(
                         btype=btype,
@@ -41,7 +42,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
                 w += workflow_pair(
                     btype=btype,
                     python_version=python_version,
-                    pytorch_version="1.5",
+                    pytorch_version="1.6.0",
                     cu_version=cu_version,
                     prefix=prefix,
                     upload=upload,