diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
index 2e84feea..deb039df 100644
--- a/.github/workflows/pr.yaml
+++ b/.github/workflows/pr.yaml
@@ -24,7 +24,7 @@ jobs:
     with:
       enable_check_generated_files: false
   wheel-build:
-    runs-on: linux-${{ matrix.arch }}-gpu-v100-latest-1
+    runs-on: linux-${{ matrix.arch }}-${{ inputs.node_type }}"
     strategy:
       matrix:
         python-version: ["3.9", "3.10"]