Skip to content

parallelizing L2 cache lookup#3032

Closed
duduyi2013 wants to merge 1 commit into
pytorch:mainfrom
duduyi2013:export-D61417947
Closed

parallelizing L2 cache lookup#3032
duduyi2013 wants to merge 1 commit into
pytorch:mainfrom
duduyi2013:export-D61417947

Conversation

@duduyi2013

Copy link
Copy Markdown
Contributor

Summary:
Change sets

  1. instead of allocate an intermediate tensor to collect the L2 cache miss info, we will do all the embeddings copy inside the originally provided tensor and mark related indices to -1
  2. paralizing the cache lookup logic using multiple cachelib pools which helps reduce the LRU contention
  3. fix cachelib->UVA tensor data copy bug(wrong offset)

Differential Revision: D61417947

@netlify

netlify Bot commented Aug 24, 2024

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs failed.

Name Link
🔨 Latest commit 4237e3a
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66cfae3fdb2e2a00086b17f1

@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

Summary:
Pull Request resolved: pytorch#3032

X-link: facebookresearch/FBGEMM#130

Change sets
1. instead of allocate an intermediate tensor to collect the L2 cache miss info, we will do all the embeddings copy inside the originally provided tensor and mark related indices to -1
2. paralizing the cache lookup logic using multiple cachelib pools which helps reduce the LRU contention
3. fix cachelib->UVA tensor data copy bug(wrong offset)

Reviewed By: ehsanardestani

Differential Revision: D61417947
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in c41d676.

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