Skip to content

ci(blas): derive host-ASAN thread budgets from the CPU allocation - #15249

Draft
rkayaith wants to merge 1 commit into
developfrom
users/rkayaith/AIHPBLAS-5081-asan-cpu-budget
Draft

rkayaith wants to merge 1 commit into
developfrom
users/rkayaith/AIHPBLAS-5081-asan-cpu-budget

Conversation

@rkayaith

@rkayaith rkayaith commented Nov 6, 2026

Copy link
Copy Markdown
Contributor

JIRA ID : AIHPBLAS-5081

Motivation

The focused hipBLASLt/hipSPARSELt HOST_ASAN workflow omits TheRock's allocation-derived CPU budget. hipSPARSELt therefore uses runtime defaults, while the hipBLASLt wrapper forces one OpenMP thread. The sparse quick selection also excludes the slow 128x128 strided-batched smoke cases.

Technical Details

Forward KUBE_CPU_REQUEST into the focused test containers and set both OMP_NUM_THREADS and OPENBLAS_NUM_THREADS to its integer CPU allocation, with a minimum of one. Missing or invalid allocation metadata preserves inherited settings. Apply the setup to both libraries on gfx942 and gfx950.

Let host-ASAN hipBLASLt inherit that budget, retaining the existing full-device-ASAN overrides. Remove the NN/NT/TN/TT 128x128 ASAN exclusions from hipSPARSELt's quick filter while retaining the known-bug exclusion. Add wrapper regression checks to the existing CPU setup job and document the runtime policy. This commit contains no C++ thread cap or rocRoller optimization.

This follows the ordinary TheRock component-job budget setup and updates the focused workflow introduced by #14240. The HOST_ASAN summary remains informational rather than a required merge check.

Test Plan

  • Validate wrapper budget propagation and restored sparse smoke coverage, including the unchanged device-ASAN overrides.
  • Check integer, fractional, missing and invalid allocation values and run pre-commit.
  • Measure the draft PR's gfx942/gfx950 CI test-step durations and outcomes against the previous configuration. Record the added sparse test coverage alongside timing; the old and new quick selections differ.
  • Separately sweep 25, 24, 16, 8, 4 and 1 threads locally on the complete regular MX group and the restored sparse families, with both variables set to the selected count.

Test Result

The wrapper regression checks fail on the parent revision for the host-ASAN budget override and sparse exclusions, and all three checks pass with this change. Eight allocation-parsing cases pass. Pre-commit passes. GPU CI timing and the local sweep are pending; this draft makes no speedup claim.

Submission Checklist

Risk level

Medium: test-client CPU concurrency changes and previously excluded sparse cases run again under ASAN. Existing sanitizer settings, test correctness checks and timeouts remain in place. hipSPARSELt has no component TESTING.md; its README documents the test executable, and the shared host-ASAN policy is documented in hipBLASLt's TESTING.md.

Pass KUBE_CPU_REQUEST into the focused hipBLASLt/hipSPARSELt HOST_ASAN
containers and set OpenMP and OpenBLAS thread budgets from its integer
CPU allocation, following TheRock component CI. Preserve inherited
settings when the allocation is unavailable or invalid.

Let host-ASAN hipBLASLt retain the configured OpenMP budget; keep the
existing full-device-ASAN overrides. Restore the 128x128 NN/NT/TN/TT
strided-batched hipSPARSELt smoke cases under ASAN.

Add launch-boundary regression checks to the workflow setup job and
document the runtime policy. The checks reproduce both issues before
the change and pass afterward; eight allocation parsing cases and
pre-commit checks pass. GPU CI timing measurements are pending.

JIRA ID : AIHPBLAS-5081
@therock-pr-bot

therock-pr-bot Bot commented Nov 6, 2026 •

Copy link
Copy Markdown

✅ All Checks Passed — Ready for Review

Check Status Details
📝 PR Description ✅ Pass —
⛔ Forbidden Files ✅ Pass —
🧪 Unit Test ✅ Pass PR does not contain code files — Unit Test auto-passed
🔎 pre-commit ✅ Pass —
🚫 Draft PR 🔜 To Be Enabled —
🚩 Feature Flag 🔜 To Be Enabled —
📊 Code Coverage 🔜 To Be Enabled —
🤖 therock-pr-bot ✅ Pass —

🎉 All checks passed! This PR is ready for review.

📖 Need help? See the Policy FAQ for details on every check and how to fix failures.

🙋 Wish to Override Policy?

@therock-pr-bot

therock-pr-bot Bot commented Nov 6, 2026

Copy link
Copy Markdown

🎉 All checks passed! This PR is ready for review.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant