Skip to content

Add CUDA GPU benchmark results#1280

Merged
s3alfisc merged 9 commits intomasterfrom
cuda_gpu_benchmarks
May 2, 2026
Merged

Add CUDA GPU benchmark results#1280
s3alfisc merged 9 commits intomasterfrom
cuda_gpu_benchmarks

Conversation

@apoorvalal
Copy link
Copy Markdown
Member

Summary

  • configure the benchmark environment for the CUDA GPU stack with torch, JAX, and CuPy support
  • add torch-cuda, JAX, and CuPy benchmark result CSVs for base, AKM sweep, and AKM occupation benchmarks
  • refresh benchmark figures and update plotting styles so full backend legends have distinct color/marker pairs

Verification

  • pixi run -e benchmark benchmark
  • pixi run -e benchmark benchmark-akm
  • pixi run -e benchmark benchmark-akm-occupation
  • PYFIXEST_BENCHMARK_INCLUDE_CUPY=1 full benchmark suite exited 0 for all three tasks
  • pixi run pytest tests/test_modular_plotting.py

@codecov
Copy link
Copy Markdown

codecov Bot commented Apr 19, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

Flag Coverage Δ
core-tests 72.83% <ø> (ø)
test-r-core 51.28% <ø> (ø)
test-r-extended 19.00% <ø> (ø)
test-r-fixest 40.56% <ø> (ø)
tests-extended ?

Flags with carried forward coverage won't be shown. Click here to find out more.

Files with missing lines Coverage Δ
pyfixest/utils/dgps.py 55.80% <ø> (ø)

... and 6 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@s3alfisc s3alfisc merged commit 4daf05a into master May 2, 2026
12 of 13 checks passed
@s3alfisc s3alfisc deleted the cuda_gpu_benchmarks branch May 2, 2026 11:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants