Add dedicated Helion workflow for Nebius B200 runners

The helion runner image uses python3 -m pip (no bare pip) and needs
sudo for apt-get. Create a separate helion_workflow.yml instead of
reusing the NVIDIA workflow.

Author: Mark Saroufim

.github/workflows/helion_workflow.yml

@@ -0,0 +1,70 @@
+name: Helion Job
+on:
+  workflow_dispatch:
+    inputs:
+      run_id:
+        description: 'Unique identifier for this run'
+        required: true
+        type: string
+      payload:
+        description: 'Content of the user submission, as json string'
+        required: true
+        type: string
+      runner:
+        description: 'Helion runner to run workflow on'
+        required: true
+        default: "nebius-b200-helion-runners"
+        type: string
+      requirements:
+        description: 'Contents for a requirements.txt file'
+        required: false
+        type: string
+
+run-name: 'Helion Job - ${{ github.event.inputs.run_id }}'
+
+jobs:
+  run:
+    runs-on: ${{ github.event.inputs.runner }}
+    timeout-minutes: 20
+    steps:
+    - uses: actions/checkout@v3
+
+    - name: Create input files
+      shell: bash
+      run: |
+        # Extract the payload content without printing it
+        sudo apt-get update && sudo apt-get install -y jq
+        PAYLOAD=$(jq -r '.inputs.payload' $GITHUB_EVENT_PATH)
+
+        # Apply mask to the extracted content
+        echo "::add-mask::$PAYLOAD"
+
+        # Now write to file (won't be logged since it's masked)
+        echo "$PAYLOAD" > payload.json
+
+    - name: Setup Virtual Environment and Install Dependencies
+      shell: bash
+      run: |
+        python3 -m pip install --upgrade pip
+        python3 -m pip install -r "requirements-dev.txt"
+        python3 -m pip install -e .
+
+    - name: Run script
+      shell: bash
+      run: |
+        python3 src/runners/github-runner.py
+
+    - name: Upload training artifacts
+      uses: actions/upload-artifact@v4
+      if: always()
+      with:
+        name: run-result
+        path: result.json
+
+    - name: Upload profiling artifacts
+      uses: actions/upload-artifact@v4
+      if: always()
+      with:
+        name: profile-data
+        path: profile_data/*
+        retention-days: 1

src/libkernelbot/launchers/github.py

@@ -105,12 +105,14 @@ async def run_submission(  # noqa: C901
             }[gpu_type.value]
             gpu_vendor = "AMD"
             requirements = AMD_REQUIREMENTS
-        elif gpu_type.value in ["NVIDIA", "B200_Nebius"]:
+        elif gpu_type.value == "B200_Nebius":
+            selected_workflow = "helion_workflow.yml"
+            runner_name = "nebius-b200-helion-runners"
+            gpu_vendor = "NVIDIA"
+            requirements = NVIDIA_REQUIREMENTS
+        elif gpu_type.value == "NVIDIA":
             selected_workflow = "nvidia_workflow.yml"
-            runner_name = {
-                "NVIDIA": "nvidia-docker-b200-8-x86-64",
-                "B200_Nebius": "nebius-b200-helion-runners",
-            }[gpu_type.value]
+            runner_name = "nvidia-docker-b200-8-x86-64"
             gpu_vendor = "NVIDIA"
             requirements = NVIDIA_REQUIREMENTS
         else:
@@ -307,6 +309,8 @@ async def trigger(self, inputs: dict) -> bool:
             expected_run_name = f"AMD Job - {run_id}"
         elif self.workflow_file == "nvidia_workflow.yml":
             expected_run_name = f"NVIDIA Job - {run_id}"
+        elif self.workflow_file == "helion_workflow.yml":
+            expected_run_name = f"Helion Job - {run_id}"
         else:
             raise ValueError(f"Unknown workflow file: {self.workflow_file}")