Skip to content

Add VBE support in UVM caching training#2388

Closed
sryap wants to merge 1 commit into
pytorch:mainfrom
sryap:export-D54512865
Closed

Add VBE support in UVM caching training#2388
sryap wants to merge 1 commit into
pytorch:mainfrom
sryap:export-D54512865

Conversation

@sryap

@sryap sryap commented Mar 5, 2024

Copy link
Copy Markdown
Contributor

Summary:
This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
linearize_cache_indices.

Differential Revision: D54512865

@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@netlify

netlify Bot commented Mar 5, 2024

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 40ce817
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/65e767c9c06a9a0008b612c7
😎 Deploy Preview https://deploy-preview-2388--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 site configuration.

sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Differential Revision: D54512865
sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Differential Revision: D54512865
sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Differential Revision: D54512865
@sryap sryap force-pushed the export-D54512865 branch from 89b993c to 1b02484 Compare March 5, 2024 18:34
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Differential Revision: D54512865
@sryap sryap force-pushed the export-D54512865 branch from 1b02484 to 49faa67 Compare March 5, 2024 18:34
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Backend: D54512865 (this diff)
Frontend: D54546141

Differential Revision: D54512865
@sryap sryap force-pushed the export-D54512865 branch from 49faa67 to b7eab7c Compare March 5, 2024 18:36
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Mar 5, 2024
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Backend: D54512865 (this diff)
Frontend: D54546141

Differential Revision: D54512865
@sryap sryap force-pushed the export-D54512865 branch from b7eab7c to a943604 Compare March 5, 2024 18:43
Summary:

This diff adds VBE (variable batch size embedding) support to TBE with
UVM caching, specifically adding the VBE support in
`linearize_cache_indices`.

Backend: D54512865 (this diff)
Frontend: D54546141

Differential Revision: D54512865
@sryap sryap force-pushed the export-D54512865 branch from a943604 to 40ce817 Compare March 5, 2024 18:43
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

1 similar comment
@facebook-github-bot

Copy link
Copy Markdown
Contributor

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

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in d3a0d04.

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