Code

Improvements to benchmarking system#248

Code

Merged

matt-graham merged 15 commits intomainastro-informatics/s2fft:mainfrom mmg/benchmark-improvementsastro-informatics/s2fft:mmg/benchmark-improvementsCopy head branch name to clipboardDec 11, 2024

+395-52

Commits

Commits on Nov 27, 2024

• Use JAX function for generating precomputes

  matt-graham

  committed
  fdf93e3

  Copy full SHA for fdf93e3

• Add benchmarks for precompute versions of spherical transforms

  matt-graham

  committed
  268e621

  Copy full SHA for 268e621

• Show defaults in benchmark help text

  matt-graham

  committed
  1f5182a

  Copy full SHA for 1f5182a

• Add flag for running once and discarding to eliminate JIT overhead

  matt-graham

  committed
  24d0441

  Copy full SHA for 24d0441

• Use module docstring for benchmark help description

  matt-graham

  committed
  3adac86

  Copy full SHA for 3adac86

• Make benchmark module docstring consistent

  matt-graham

  committed
  0c87cd6

  Copy full SHA for 0c87cd6

• Add precompute Wigner benchmarks

  matt-graham

  committed
  b3e3283

  Copy full SHA for b3e3283

• Record all benchmark results to JSON to not just last param set

  matt-graham

  committed
  dfaa569

  Copy full SHA for dfaa569

• Record system info and date time in JSON output

  matt-graham

  committed
  54db6d2

  Copy full SHA for 54db6d2

• Expose recursion and mode parameters for precompute benchmarks

  matt-graham

  committed
  f4e9458

  Copy full SHA for f4e9458

• Record CPU and GPU info in benchmarks output if available

  matt-graham

  committed
  e917f05

  Copy full SHA for e917f05

• Record GPU memory + CUDA info in results

  matt-graham

  committed
  742fdad

  Copy full SHA for 742fdad

• Fix Wigner benchmarks

  matt-graham

  committed
  942cf9e

  Copy full SHA for 942cf9e

Commits on Nov 29, 2024

• Removing unused imports

  matt-graham

  committed
  64a2c71

  Copy full SHA for 64a2c71

• Update benchmarking README to reflect new modules and functionality

  matt-graham

  committed
  77fa3a3

  Copy full SHA for 77fa3a3