Fix naming

Author: atalman

.github/workflows/validate-domain-library.yml

@@ -1,4 +1,4 @@
-name: Validate binaries
+name: Validate domain libary
 
 # A reusable workflow that triggers a set of jobs that perform a smoke test / validation of pytorch binaries.
 # Optionally restricts validation to the specified OS and channel.
@@ -22,59 +22,76 @@ on:
         default: ""
         required: false
         type: string
-  workflow_dispatch:
-    inputs:
-      os:
-        description: "Operating system to generate for (linux, windows, macos, macos-arm64)"
+      package-type:
+        description: "Package type (conda, wheel, all)"
         required: true
-        type: choice
-        default: all
-        options:
-          - windows
-          - linux
-          - macos
-          - all
-      channel:
-        description: "Channel to use (nightly, test, release, all)"
+        type: string
+      repository:
+        description: "Path to repository to checkout"
+        required: true
+        type: string
+      smoke_test:
+        description: "Path to a smoke test script"
         required: true
-        type: choice
-        default: all
-        options:
-          - release
-          - nightly
-          - test
-          - all
-      ref:
-        description: 'Reference to checkout, defaults to empty'
-        default: ""
-        required: false
         type: string
 
 jobs:
-  win:
-    if:  inputs.os == 'windows' || inputs.os == 'all'
-    uses: ./.github/workflows/validate-windows-binaries.yml
-    with:
-      channel: ${{ inputs.channel }}
-      ref: ${{ inputs.ref || github.ref }}
 
-  linux:
-    if:  inputs.os == 'linux' || inputs.os == 'all'
-    uses: ./.github/workflows/validate-linux-binaries.yml
+  generate-linux-conda-matrix:
+    if:  (inputs.os == 'linux' || inputs.os == 'all') && (inputs.package_type == 'conda' || inputs.package_type == 'all')
+    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
     with:
+      package-type: conda
+      os: linux
       channel: ${{ inputs.channel }}
-      ref: ${{ inputs.ref || github.ref }}
-
-  mac:
-    if:  inputs.os == 'macos' || inputs.os == 'all'
-    uses: ./.github/workflows/validate-macos-binaries.yml
+      with-cuda: disable
+  generate-linux-wheel-matrix:
+    if:  (inputs.os == 'linux' || inputs.os == 'all') && (inputs.package_type == 'wheel' || inputs.package_type == 'all')
+    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
     with:
-      channel: ${{ inputs.channel }}
+      package-type: wheel
+      os: linux
+      channel: ${{ inputs.channel || 'nightly' }}
+      with-cuda: disable
+  validate-conda-linux:
+    if:  (inputs.os == 'linux' || inputs.os == 'all') && (inputs.package_type == 'conda' || inputs.package_type == 'all')
+    needs: generate-linux-conda-matrix
+    strategy:
+      matrix: ${{ fromJson(needs.generate-linux-conda-matrix.outputs.matrix) }}
+      fail-fast: false
+    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+    name: "text-${{ matrix.package_type }}-${{ matrix.python_version }}"
+    with:
+      runner: "linux.2xlarge"
+      repository: ${{ input.repository }}
       ref: ${{ inputs.ref || github.ref }}
-
-  mac-arm64:
-    if:  inputs.os == 'macos' || inputs.os == 'all'
-    uses: ./.github/workflows/validate-macos-arm64-binaries.yml
+      job-name: "text-${{ matrix.package_type }}-${{ matrix.python_version }}"
+      script: |
+        set -ex
+        export ENV_NAME="conda-env-${{ github.run_id }}"
+        export DESIRED_PYTHON="${{ matrix.python_version }}"
+        export PACKAGE_TYPE="${{ matrix.package_type }}"
+        export CHANNEL="${{ matrix.channel }}"
+        export SMOKE_TEST="${{ input.smoke_test }}"
+        eval $SMOKE_TEST
+  validate-wheel-linux:
+    if: (inputs.os == 'linux' || inputs.os == 'all') && (inputs.package_type == 'wheel' || inputs.package_type == 'all')
+    needs: generate-linux-wheel-matrix
+    strategy:
+      matrix: ${{ fromJson(needs.generate-linux-wheel-matrix.outputs.matrix) }}
+      fail-fast: false
+    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+    name: "text-${{ matrix.package_type }}-${{ matrix.python_version }}"
     with:
-      channel: ${{ inputs.channel }}
+      runner: "linux.2xlarge"
+      repository: ${{ input.repository }}
       ref: ${{ inputs.ref || github.ref }}
+      job-name: "text-${{ matrix.package_type }}-${{ matrix.python_version }}"
+      script: |
+        set -ex
+        export ENV_NAME="conda-env-${{ github.run_id }}"
+        export DESIRED_PYTHON="${{ matrix.python_version }}"
+        export PACKAGE_TYPE="${{ matrix.package_type }}"
+        export CHANNEL="${{ matrix.channel }}"
+        export SMOKE_TEST="${{ input.smoke_test }}"
+        eval $SMOKE_TEST