From 5dd14492c52da39e0e9f78c0802ce77ee1b15446 Mon Sep 17 00:00:00 2001
From: PaliC <sahancpal@gmail.com>
Date: Thu, 21 Mar 2024 10:42:30 -0700
Subject: [PATCH] add versioning

---
 .github/workflows/validate-binaries.yml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.github/workflows/validate-binaries.yml b/.github/workflows/validate-binaries.yml
index 6bebfd3fcb..fe55276dd8 100644
--- a/.github/workflows/validate-binaries.yml
+++ b/.github/workflows/validate-binaries.yml
@@ -43,11 +43,17 @@ on:
         default: ""
         required: false
         type: string
+      pytorch_version:
+        description: 'PyTorch version to validate (ie. 2.0, 2.2.2, etc.) - optional'
+        default: ""
+        required: false
+        type: string
 jobs:
   validate-binaries:
     uses: pytorch/test-infra/.github/workflows/validate-domain-library.yml@main
     with:
       package_type: "conda,wheel"
+      version: ${{ inputs.version }}
       os: ${{ inputs.os }}
       channel: ${{ inputs.channel }}
       repository: "pytorch/text"