Skip to content
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

Create an interface to enable eviction policy #2479

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

dracifer
Copy link
Contributor

Summary:
To support various types of eviction policy, the HashZchManagedCollisionModule needs to be able to calculate a score (e.g., TTL) for each incoming ID and pass it to the kernel. The latter will make an informed eviction decision based on the existing score associated with the identity (stored in metadata), and a reference value (e.g., current timestamp). Refer to Doc for details.

This Diff uses per-feature TTL as an example, and creates a scorer to assign a TTL for each ID with regard to the expiration hr of the feature. The score will be passed to the kernel (in a separate Diff) to perform eviction.

Differential Revision: D63879449

@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 Oct 10, 2024
@facebook-github-bot
Copy link
Contributor

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

Summary:
Pull Request resolved: pytorch#2478

To support per-feature TTL, feature split needs to be passed to the ZCH module so features that share the same embedding table can be differentiated.

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

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

dracifer pushed a commit to dracifer/torchrec that referenced this pull request Oct 10, 2024
Summary:
Pull Request resolved: pytorch#2479

To support various types of eviction policy, the `HashZchManagedCollisionModule` needs to be able to calculate a score (e.g., TTL) for each incoming ID and pass it to the kernel. The latter will make an informed eviction decision based on the existing score associated with the identity (stored in metadata), and a reference value (e.g., current timestamp). Refer to [Doc](https://docs.google.com/document/d/19s6N4-CWhveqhV1ePwrHAF8SIlzePsoxKElPRnnlo8c/edit?usp=sharing) for details.

This Diff uses per-feature TTL as an example, and creates a scorer to assign a TTL for each ID with regard to the expiration hr of the feature. The score will be passed to the kernel (in a separate Diff) to perform eviction.

Differential Revision: D63879449
Summary:
Pull Request resolved: pytorch#2479

To support various types of eviction policy, the `HashZchManagedCollisionModule` needs to be able to calculate a score (e.g., TTL) for each incoming ID and pass it to the kernel. The latter will make an informed eviction decision based on the existing score associated with the identity (stored in metadata), and a reference value (e.g., current timestamp). Refer to [Doc](https://docs.google.com/document/d/19s6N4-CWhveqhV1ePwrHAF8SIlzePsoxKElPRnnlo8c/edit?usp=sharing) for details.

This Diff uses per-feature TTL as an example, and creates a scorer to assign a TTL for each ID with regard to the expiration hr of the feature. The score will be passed to the kernel (in a separate Diff) to perform eviction.

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

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

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