Skip to content

Account for cache load factor in memory estimate #3035

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from

Conversation

kausv
Copy link
Contributor

@kausv kausv commented Jun 3, 2025

Summary:
The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Differential Revision: D75893983

@facebook-github-bot facebook-github-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 Jun 3, 2025
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75893983

kausv added a commit to kausv/torchrec that referenced this pull request Jun 3, 2025
Summary:

The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Reviewed By: emlin

Differential Revision: D75893983
@kausv kausv force-pushed the export-D75893983 branch from 5f53ff3 to b1f3b50 Compare June 3, 2025 20:09
kausv added a commit to kausv/torchrec that referenced this pull request Jun 3, 2025
Summary:

The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Reviewed By: emlin

Differential Revision: D75893983
@kausv kausv force-pushed the export-D75893983 branch from b1f3b50 to 40a0db4 Compare June 3, 2025 20:11
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75893983

kausv added a commit to kausv/torchrec that referenced this pull request Jun 3, 2025
Summary:
Pull Request resolved: pytorch#3035

The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Reviewed By: emlin

Differential Revision: D75893983
@kausv kausv force-pushed the export-D75893983 branch from 40a0db4 to bb323b2 Compare June 3, 2025 20:15
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75893983

kausv added a commit to kausv/torchrec that referenced this pull request Jun 3, 2025
Summary:
Pull Request resolved: pytorch#3035

The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Reviewed By: emlin

Differential Revision: D75893983
@kausv kausv force-pushed the export-D75893983 branch from bb323b2 to ab7b7c3 Compare June 3, 2025 20:20
kaus and others added 2 commits June 3, 2025 14:07
Summary: the variable is kept optional str for "inf"

Differential Revision: D75900311
Summary:
Pull Request resolved: pytorch#3035

The actual L1 cache allocated is determined by cache load factor and ceiled by max_l1_cache_size config.
This diff addresses that to improve the memory estimate.

Reviewed By: emlin, duduyi2013

Differential Revision: D75893983
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75893983

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
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants