setting up runs-on as a parameter to check-go-mod

.github/workflows/build-distros.yml

@@ -8,8 +8,26 @@ permissions:
   contents: read
 
 jobs:
+  setup:
+    name: Setup
+    runs-on: ubuntu-latest
+    outputs:
+      compute-small: ${{ steps.setup-outputs.outputs.compute-small }}
+      compute-medium: ${{ steps.setup-outputs.outputs.compute-medium }}
+      compute-large: ${{ steps.setup-outputs.outputs.compute-medium }}
+      compute-xl: ${{ steps.setup-outputs.outputs.compute-xl }}
+    steps:
+    - uses: actions/checkout@ac593985615ec2ede58e132d2e21d2b1cbd6127c # [email protected]
+    - id: setup-outputs
+      name: Setup outputs
+      run: ./.github/scripts/get_runner_classes.sh
+
   check-go-mod:
+    needs: 
+    - setup
     uses: ./.github/workflows/reusable-check-go-mod.yml
+    with:
+      runs-on: ${{ needs.setup.outputs.compute-medium }}
 
   build-386:
     needs: check-go-mod

.github/workflows/reusable-check-go-mod.yml

@@ -2,10 +2,19 @@ name: check-go-mod
 
 on:
   workflow_call:
-
+    inputs:
+      uploaded-binary-name:
+        required: false
+        type: string
+        default: "consul-bin"
+      runs-on:
+        description: An expression indicating which kind of runners to use.
+        required: false
+        type: string
+        default: '["custom", "linux", "small"]'
 jobs:
   check-go-mod:
-    runs-on: ubuntu-latest
+    runs-on: ${{ fromJSON(inputs.runs-on) }}
 
     steps:
     - uses: actions/checkout@ac593985615ec2ede58e132d2e21d2b1cbd6127c # [email protected]