Skip to content

Commit

Permalink
add telemetry (#321)
Browse files Browse the repository at this point in the history
Adds telemetry that will send timing information about each step in the build process to a time series database, for later viewing and analytics.

Tracking issue: rapidsai/build-infra#139

This is not expected to work right now. We are in flux regarding a time series database that is in the same private network as the build machines. @ajschmidt8 has restricted the mTLS certificates that are used for the current database to specific repos. Either he will enable those here, or we'll make the switch to the new server, and then this mechanism should work.

Authors:
  - Mike Sarahan (https://github.com/msarahan)
  - James Lamb (https://github.com/jameslamb)

Approvers:
  - James Lamb (https://github.com/jameslamb)

URL: #321
  • Loading branch information
msarahan authored Feb 12, 2025
1 parent 050b66d commit 35d4903
Showing 1 changed file with 27 additions and 1 deletion.
28 changes: 27 additions & 1 deletion .github/workflows/pr.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ jobs:
- wheel-tests-ucxx
- wheel-build-distributed-ucxx
- wheel-tests-distributed-ucxx
- telemetry-setup
secrets: inherit
uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-25.04
if: always()
Expand All @@ -43,8 +44,21 @@ jobs:
uses: rapidsai/shared-actions/check_nightly_success/dispatch@main
with:
repo: ucxx

telemetry-setup:
runs-on: ubuntu-latest
continue-on-error: true
env:
OTEL_SERVICE_NAME: "pr-ucxx"
steps:
- name: Telemetry setup
# This gate is here and not at the job level because we need the job to not be skipped,
# since other jobs depend on it.
if: ${{ vars.TELEMETRY_ENABLED == 'true' }}
uses: rapidsai/shared-actions/telemetry-dispatch-stash-base-env-vars@main
changed-files:
secrets: inherit
needs: telemetry-setup
uses: rapidsai/shared-workflows/.github/workflows/changed-files.yaml@branch-25.04
with:
files_yaml: |
Expand All @@ -64,8 +78,10 @@ jobs:
checks:
secrets: inherit
uses: rapidsai/shared-workflows/.github/workflows/checks.yaml@branch-25.04
needs: telemetry-setup
with:
enable_check_generated_files: false
ignored_pr_jobs: telemetry-summarize
conda-cpp-build:
needs: checks
secrets: inherit
Expand Down Expand Up @@ -150,7 +166,7 @@ jobs:
script: ci/test_wheel_distributed_ucxx.sh
devcontainer:
secrets: inherit
needs: checks
needs: [checks, telemetry-setup]
uses: rapidsai/shared-workflows/.github/workflows/build-in-devcontainer.yaml@branch-25.04
with:
arch: '["amd64"]'
Expand All @@ -159,3 +175,13 @@ jobs:
sccache -z;
build-all --verbose;
sccache -s;
telemetry-summarize:
# This job must use a self-hosted runner to record telemetry traces.
runs-on: linux-amd64-cpu4
needs: pr-builder
if: ${{ vars.TELEMETRY_ENABLED == 'true' && !cancelled() }}
continue-on-error: true
steps:
- name: Telemetry summarize
uses: rapidsai/shared-actions/telemetry-dispatch-summarize@main

0 comments on commit 35d4903

Please sign in to comment.