Skip to content

Add SSD backend integration and metadata API to DramKVEmbeddingCache (#5929)#5929

Closed
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D108959007
Closed

Add SSD backend integration and metadata API to DramKVEmbeddingCache (#5929)#5929
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D108959007

Conversation

@lizhe-ji

@lizhe-ji lizhe-ji commented Jun 18, 2026

Copy link
Copy Markdown

Summary:

X-link: https://github.com/facebookresearch/FBGEMM/pull/2847

Integrates DramKVEmbeddingCache with an SSD backend by exposing metadata retrieval APIs and internal state accessors. This enables the SSD tier to track dirty memory blocks for flushing, manage cross-tier feature eviction, and allows the enrichment process to skip redundant external data source fetches for IDs already present in SSD.

Differential Revision: D108959007

@meta-cla meta-cla Bot added the cla signed label Jun 18, 2026
@lizhe-ji lizhe-ji force-pushed the export-D108959007 branch 2 times, most recently from fa1da14 to 0cdbc89 Compare June 19, 2026 01:51
lizhe-ji pushed a commit to lizhe-ji/FBGEMM-1 that referenced this pull request Jun 19, 2026
…ytorch#5929)

Summary:

X-link: facebookresearch/FBGEMM#2847

Integrates `DramKVEmbeddingCache` with an SSD backend by exposing metadata retrieval APIs and internal state accessors. This enables the SSD tier to track dirty memory blocks for flushing, manage cross-tier feature eviction, and allows the enrichment process to skip redundant external data source fetches for IDs already present in SSD.

Differential Revision: D108959007
@meta-codesync meta-codesync Bot changed the title Add SSD backend integration and metadata API to DramKVEmbeddingCache Add SSD backend integration and metadata API to DramKVEmbeddingCache (#5929) Jun 19, 2026
lizhe-ji pushed a commit to lizhe-ji/FBGEMM-1 that referenced this pull request Jun 19, 2026
…ytorch#5929)

Summary:

X-link: facebookresearch/FBGEMM#2847

Integrates `DramKVEmbeddingCache` with an SSD backend by exposing metadata retrieval APIs and internal state accessors. This enables the SSD tier to track dirty memory blocks for flushing, manage cross-tier feature eviction, and allows the enrichment process to skip redundant external data source fetches for IDs already present in SSD.

Differential Revision: D108959007
…ytorch#5929)

Summary:

X-link: facebookresearch/FBGEMM#2847

Integrates `DramKVEmbeddingCache` with an SSD backend by exposing metadata retrieval APIs and internal state accessors. This enables the SSD tier to track dirty memory blocks for flushing, manage cross-tier feature eviction, and allows the enrichment process to skip redundant external data source fetches for IDs already present in SSD.

Differential Revision: D108959007
@lizhe-ji lizhe-ji force-pushed the export-D108959007 branch from 0cdbc89 to 4d3d5d8 Compare June 22, 2026 23:54
lizhe-ji pushed a commit to lizhe-ji/FBGEMM-1 that referenced this pull request Jun 23, 2026
…ytorch#5929)

Summary:

X-link: facebookresearch/FBGEMM#2847

Integrates `DramKVEmbeddingCache` with an SSD backend by exposing metadata retrieval APIs and internal state accessors. This enables the SSD tier to track dirty memory blocks for flushing, manage cross-tier feature eviction, and allows the enrichment process to skip redundant external data source fetches for IDs already present in SSD.

Reviewed By: EddyLXJ

Differential Revision: D108959007
@meta-codesync meta-codesync Bot closed this in 93221e9 Jun 23, 2026
@meta-codesync meta-codesync Bot added the Merged label Jun 23, 2026
@meta-codesync

meta-codesync Bot commented Jun 23, 2026

Copy link
Copy Markdown
Contributor

This pull request has been merged in 93221e9.

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