Update action.yml

.github/actions/inductor-test/action.yml

@@ -1,5 +1,4 @@
 name: Test build/test linux gpu
-
 on:
   push:
     branches: [ master, main ]
@@ -9,10 +8,12 @@ on:
         description: 'Triton branch or commit to pin'
         default: 'main'
         required: false
+        type: string
       pytorch_pin:
         description: 'PyTorch branch or commit to pin'
         default: 'main'
         required: false
+        type: string
 
 jobs:
   build-test:
@@ -22,15 +23,14 @@ jobs:
       gpu-arch-type: cuda
       gpu-arch-version: "12.1"
       timeout: 360
-      # docker-image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
       script: |
         set -x
         pushd ..
         echo "Installing triton"
         git clone https://github.com/triton-lang/triton.git
         pushd triton
         echo "Checking out triton branch or commit"
-        git checkout ${{ inputs.triton_pin || 'main' }}
+        git checkout ${{ github.event.inputs.triton_pin || 'main' }}
         sudo yum install -y zlib-devel
         echo "Installing build-time dependencies"
         pip install ninja==1.11.1.1 cmake==3.30.2 wheel==0.44.0
@@ -58,7 +58,7 @@ jobs:
         git clone https://github.com/pytorch/pytorch.git
         pushd pytorch
         echo "Checking out pytorch branch or commit"
-        git checkout ${{ inputs.pytorch_pin || 'main' }}
+        git checkout ${{ github.event.inputs.pytorch_pin || 'main' }}
         git submodule sync
         git submodule update --init --recursive
         pip install -r requirements.txt