Skip to content

Adding new ID_COUNT eviction trigger condition#4829

Closed
EddyLXJ wants to merge 1 commit into
pytorch:mainfrom
EddyLXJ:export-D81151216
Closed

Adding new ID_COUNT eviction trigger condition#4829
EddyLXJ wants to merge 1 commit into
pytorch:mainfrom
EddyLXJ:export-D81151216

Conversation

@EddyLXJ

@EddyLXJ EddyLXJ commented Sep 5, 2025

Copy link
Copy Markdown
Contributor

Differential Revision: D81151216

@netlify

netlify Bot commented Sep 5, 2025

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit bbfb1a2
🔍 Latest deploy log https://app.netlify.com/projects/pytorch-fbgemm-docs/deploys/68c35fa43c1a200008368ff1
😎 Deploy Preview https://deploy-preview-4829--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify project configuration.

@meta-cla meta-cla Bot added the cla signed label Sep 5, 2025
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

EddyLXJ added a commit to EddyLXJ/FBGEMM-1 that referenced this pull request Sep 5, 2025
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

EddyLXJ added a commit to EddyLXJ/torchrec that referenced this pull request Sep 5, 2025
Summary:
X-link: meta-pytorch/torchrec#3357


X-link: facebookresearch/FBGEMM#1855


ID COUNT Trigger is based on total num of id in all kvzch tables. Memory usage is cross all tables but eviction trigger is for tbe, like:
if target total memory is 10G tbe1 has table [table0, table1], training_id_eviction_trigger_count is [200, 300] tbe2 has table [table2], training_id_eviction_trigger_count is [500] then the target memory for table0 is 2G, table1 is 3G, table2 is 5G, but the trigger condition is tbe level.  If currently total id num in tbe1 is [100, 350], in this case tbe1 will not trigger evict even num in table1 is greater than 300 because total num. 100 + 350 = 450 < 500. If total id num in tbe1 is [100, 450], it will trigger eviction on tbe1 only for table1, because num in table0 100 < 200.

Differential Revision: D81151216
@facebook-github-bot

Copy link
Copy Markdown
Contributor

@EddyLXJ has exported this pull request. If you are a Meta employee, you can view the originating diff in D81151216.

EddyLXJ added a commit to EddyLXJ/torchrec that referenced this pull request Sep 11, 2025
Summary:

X-link: pytorch/FBGEMM#4829

X-link: facebookresearch/FBGEMM#1855


ID COUNT Trigger is based on total num of id in all kvzch tables. Memory usage is cross all tables but eviction trigger is for tbe, like:
if target total memory is 10G tbe1 has table [table0, table1], training_id_eviction_trigger_count is [200, 300] tbe2 has table [table2], training_id_eviction_trigger_count is [500] then the target memory for table0 is 2G, table1 is 3G, table2 is 5G, but the trigger condition is tbe level.  If currently total id num in tbe1 is [100, 350], in this case tbe1 will not trigger evict even num in table1 is greater than 300 because total num. 100 + 350 = 450 < 500. If total id num in tbe1 is [100, 450], it will trigger eviction on tbe1 only for table1, because num in table0 100 < 200.

Differential Revision: D81151216
facebook-github-bot pushed a commit to meta-pytorch/torchrec that referenced this pull request Sep 12, 2025
Summary:
Pull Request resolved: #3357

X-link: pytorch/FBGEMM#4829

X-link: facebookresearch/FBGEMM#1855

ID COUNT Trigger is based on total num of id in all kvzch tables. Memory usage is cross all tables but eviction trigger is for tbe, like:
if target total memory is 10G tbe1 has table [table0, table1], training_id_eviction_trigger_count is [200, 300] tbe2 has table [table2], training_id_eviction_trigger_count is [500] then the target memory for table0 is 2G, table1 is 3G, table2 is 5G, but the trigger condition is tbe level.  If currently total id num in tbe1 is [100, 350], in this case tbe1 will not trigger evict even num in table1 is greater than 300 because total num. 100 + 350 = 450 < 500. If total id num in tbe1 is [100, 450], it will trigger eviction on tbe1 only for table1, because num in table0 100 < 200.

Reviewed By: emlin

Differential Revision: D81151216

fbshipit-source-id: 693bd47e950cc63498b4cdfc7691636197c832dc
@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in b94d1ba.

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.

2 participants