Skip to content

feat: add per-table table_constraints event (#4077) - #4162

Closed
ahmed-shuaibi wants to merge 2 commits into
meta-pytorch:mainfrom
ahmed-shuaibi:export-D98000171
Closed

ahmed-shuaibi wants to merge 2 commits into
meta-pytorch:mainfrom
ahmed-shuaibi:export-D98000171

Conversation

@ahmed-shuaibi

Copy link
Copy Markdown

Summary:

Log the parameter constraints passed to the planner per table (compute_kernels, sharding_types, initial CLF, enforce_hbm, SSD offloading). Emitted right after planner_config, before enumerate(), from both OSS and LP planners.

Reviewed By: eugeneshulgameta, hammad45

Differential Revision: D98000171

Ahmed Shuaibi added 2 commits April 23, 2026 09:12
Summary:

Log the planner's final per-table assignment (compute_kernel, sharding_type, CLF, HBM/DDR) after planning completes. One Scuba event per table in best_plan, called from both OSS and LP planners next to log_offloading_summary.

Reviewed By: hammad45, eugeneshulgameta

Differential Revision: D97985051
Summary:

Log the parameter constraints passed to the planner per table (compute_kernels, sharding_types, initial CLF, enforce_hbm, SSD offloading). Emitted right after planner_config, before enumerate(), from both OSS and LP planners.

Reviewed By: eugeneshulgameta, hammad45

Differential Revision: D98000171
@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 23, 2026
@meta-codesync

meta-codesync Bot commented Apr 23, 2026

Copy link
Copy Markdown
Contributor

@ahmed-shuaibi has exported this pull request. If you are a Meta employee, you can view the originating Diff in D98000171.

ahmed-shuaibi pushed a commit to ahmed-shuaibi/torchrec that referenced this pull request Apr 23, 2026
Summary:

Log the parameter constraints passed to the planner per table (compute_kernels, sharding_types, initial CLF, enforce_hbm, SSD offloading). Emitted right after planner_config, before enumerate(), from both OSS and LP planners.

Reviewed By: eugeneshulgameta, hammad45

Differential Revision: D98000171
ahmed-shuaibi pushed a commit to ahmed-shuaibi/torchrec that referenced this pull request Apr 23, 2026
Summary:

Log the parameter constraints passed to the planner per table (compute_kernels, sharding_types, initial CLF, enforce_hbm, SSD offloading). Emitted right after planner_config, before enumerate(), from both OSS and LP planners.

Reviewed By: eugeneshulgameta, hammad45

Differential Revision: D98000171
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant