From ce9549ac717dcb932171db596bf6535bc5397de8 Mon Sep 17 00:00:00 2001 From: chengzeyi Date: Tue, 14 Nov 2023 19:10:19 +0800 Subject: [PATCH] add cudnn_version_major input for workflow_call in wheels_build.yml --- .github/workflows/wheels_build.yml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml index 4f4f4f8..0059770 100644 --- 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: