Add optional venv to upload-benchmark-results (#7083)

Helion fails with
https://github.com/pytorch/helion/actions/runs/17417766895/job/49449905513?pr=544

Author: oulgen

.github/actions/upload-benchmark-results/action.yml

@@ -10,6 +10,10 @@ inputs:
     default: 'v3'
   github-token:
     default: ''
+  venv:
+    description: 'Path to virtual environment to activate'
+    required: false
+    default: ''
 
 runs:
   using: composite
@@ -18,6 +22,10 @@ runs:
       shell: bash
       run: |
         set -eux
+
+        if [[ -n "${{ inputs.venv }}" ]]; then
+          source "${{ inputs.venv }}"
+        fi
         python3 -mpip install boto3==1.35.33 psutil==7.0.0 pynvml==12.0.0
 
         DEVICE_NAME=""
@@ -74,6 +82,10 @@ runs:
       run: |
         set -eux
 
+        if [[ -n "${{ inputs.venv }}" ]]; then
+          source "${{ inputs.venv }}"
+        fi
+
         python3 "${GITHUB_ACTION_PATH}/../../scripts/benchmarks/gather_metadata.py" \
           --schema-version "${SCHEMA_VERSION}" \
           --repo "${REPO}" \
@@ -90,6 +102,10 @@ runs:
       run: |
         set -eux
 
+        if [[ -n "${{ inputs.venv }}" ]]; then
+          source "${{ inputs.venv }}"
+        fi
+
         python3 "${GITHUB_ACTION_PATH}/../../scripts/benchmarks/gather_runners_info.py"
 
     - name: Gather the dependencies information
@@ -113,6 +129,10 @@ runs:
       run: |
         set -eux
 
+        if [[ -n "${{ inputs.venv }}" ]]; then
+          source "${{ inputs.venv }}"
+        fi
+
         if [[ ! -d "${BENCHMARK_RESULTS_DIR}" ]]; then
           echo "${BENCHMARK_RESULTS_DIR} does not exist, skipping"
           # We don't want the job to fail if the directory doesn't exist