builds for PyTorch 1.9.1

Summary: Add conda builds for the newly released PyTorch version 1.9.1.

Reviewed By: patricklabatut

Differential Revision: D31140206

fbshipit-source-id: 697549a3ef0db8248f4f9b5c00cf1407296b5022

.circleci/config.yml

@@ -368,6 +368,18 @@ workflows:
           name: linux_conda_py36_cu111_pyt190
           python_version: '3.6'
           pytorch_version: 1.9.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu102
+          name: linux_conda_py36_cu102_pyt191
+          python_version: '3.6'
+          pytorch_version: 1.9.1
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py36_cu111_pyt191
+          python_version: '3.6'
+          pytorch_version: 1.9.1
       - binary_linux_conda:
           context: DOCKERHUB_TOKEN
           cu_version: cu92
@@ -506,6 +518,18 @@ workflows:
           name: linux_conda_py37_cu111_pyt190
           python_version: '3.7'
           pytorch_version: 1.9.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu102
+          name: linux_conda_py37_cu102_pyt191
+          python_version: '3.7'
+          pytorch_version: 1.9.1
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py37_cu111_pyt191
+          python_version: '3.7'
+          pytorch_version: 1.9.1
       - binary_linux_conda:
           context: DOCKERHUB_TOKEN
           cu_version: cu92
@@ -644,6 +668,18 @@ workflows:
           name: linux_conda_py38_cu111_pyt190
           python_version: '3.8'
           pytorch_version: 1.9.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu102
+          name: linux_conda_py38_cu102_pyt191
+          python_version: '3.8'
+          pytorch_version: 1.9.1
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py38_cu111_pyt191
+          python_version: '3.8'
+          pytorch_version: 1.9.1
       - binary_linux_conda:
           context: DOCKERHUB_TOKEN
           cu_version: cu101
@@ -710,6 +746,18 @@ workflows:
           name: linux_conda_py39_cu111_pyt190
           python_version: '3.9'
           pytorch_version: 1.9.0
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu102
+          name: linux_conda_py39_cu102_pyt191
+          python_version: '3.9'
+          pytorch_version: 1.9.1
+      - binary_linux_conda:
+          context: DOCKERHUB_TOKEN
+          cu_version: cu111
+          name: linux_conda_py39_cu111_pyt191
+          python_version: '3.9'
+          pytorch_version: 1.9.1
       - binary_linux_conda_cuda:
           name: testrun_conda_cuda_py36_cu101_pyt151
           context: DOCKERHUB_TOKEN

.circleci/regenerate.py

@@ -27,6 +27,7 @@ CONDA_CUDA_VERSIONS = {
     "1.8.0": ["cu101", "cu102", "cu111"],
     "1.8.1": ["cu101", "cu102", "cu111"],
     "1.9.0": ["cu102", "cu111"],
+    "1.9.1": ["cu102", "cu111"],
 }
 
 

packaging/linux_wheels/inside.sh

@@ -38,6 +38,7 @@ declare -A CONDA_CUDA_VERSIONS=(
 #    ["1.8.0"]="cu101 cu102 cu111"
 #    ["1.8.1"]="cu101 cu102 cu111"
     ["1.9.0"]="cu102 cu111"
+    ["1.9.1"]="cu102 cu111"
 )