Skip to content

Add CPU runs to benchmark pipeline #1218

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 10 commits into
base: main
Choose a base branch
from
Draft

Conversation

imreddyTeja
Copy link
Member

@imreddyTeja imreddyTeja commented Jul 9, 2025

The benchmarks are not currently run on cpu.
Because they are only run on gpu, no flame graphs are produced.

This also adds two benchmarks using the NaN callback, but I'm
not sure how necessary that is.

Also add a flag to the Bucket benchmark,
which enabled the NaN callback. This also
adds a bucket with nan cb benchmark with cpu
and gpu to the pipeline

Add cpu benchmarks and NaN cb bench

bugfixes

import cuda

Add nan_cb tests

pickup artifacts

Purpose

To-do

Content


  • I have read and checked the items on the review checklist.

Also add a flag to the Bucket benchmark,
which enabled the NaN callback. This also
adds a bucket with nan cb benchmark with cpu
and gpu to the pipeline

Add cpu benchmarks and NaN cb bench

bugfixes

import cuda

Add nan_cb tests

pickup artifacts
@imreddyTeja imreddyTeja added the Run benchmarks Add this label to run benchmarks on clima label Jul 9, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Run benchmarks Add this label to run benchmarks on clima
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant