Skip to content

Roofline: classify LATENCY_BOUND for small global-memory transfers - #962

Draft
ajassani wants to merge 1 commit into
mainfrom
ajassani/roofline-latency-bound
Draft

Roofline: classify LATENCY_BOUND for small global-memory transfers#962
ajassani wants to merge 1 commit into
mainfrom
ajassani/roofline-latency-bound

Conversation

@ajassani

Copy link
Copy Markdown
Collaborator

Summary

  • Extend roofline classification with a global-memory latency floor (mem_latency_us in GPU arch JSON) so small transfers can be tagged LATENCY_BOUND instead of being lumped into bandwidth-bound MEMORY_BOUND.
  • Preserve legacy behavior when mem_latency_us is absent: bandwidth-only COMPUTE_BOUND / MEMORY_BOUND with a one-time UserWarning at TreePerfAnalyzer construction.
  • Document the optional arch field in examples/gpu_arch_example.md and add unit/integration tests.

Test plan

  • pytest tests/test_roofline_latency_bound.py
  • pytest tests/test_roofline_bound.py
  • Validate crossover on a real trace with mem_latency_us: 0.3` in arch JSON

@ajassani
ajassani marked this pull request as draft August 24, 2026 21:04
@codecov-commenter

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 96.87500% with 1 line in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
TraceLens/TreePerf/tree_perf.py 96.87% 1 Missing ⚠️

📢 Thoughts on this report? Let us know!

Extend global-memory roofline classification with a latency floor on
small transfers. Legacy arch files without mem_latency_us keep the old
bandwidth-only COMPUTE_BOUND/MEMORY_BOUND behavior and emit a one-time
UserWarning when TreePerfAnalyzer is constructed.

Co-authored-by: Cursor <cursoragent@cursor.com>
@ajassani
ajassani force-pushed the ajassani/roofline-latency-bound branch from 00c9449 to df25267 Compare August 24, 2026 21:30
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