add cudnn_version_major input for workflow_call in wheels_build.yml

chengzeyi · Nov 14, 2023 · ce9549a · ce9549a
1 parent 655ae7c
commit ce9549a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml
@@ -17,6 +17,11 @@ on:
         required: true
         type: string
         description: "Example: 117 for 11.7"
+      cudnn_version_major:
+        required: false
+        default: '8'
+        type: string
+        description: "Example: 8
   workflow_dispatch:
     inputs:
       os: