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

Optimize the cache fetch for forward split, pt. 2B #2282

Closed
wants to merge 1 commit into from

Conversation

q10
Copy link
Contributor

@q10 q10 commented Jan 24, 2024

Summary:
This follows up the work on D51865590 and D52679387 by plumbing the uvm_cache_stats argument passing up to the Python API level. local_uvm_cache_stats is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916

Copy link

netlify bot commented Jan 24, 2024

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 1299965
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/65c28062f909f600086559f1
😎 Deploy Preview https://deploy-preview-2282--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.

@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Jan 24, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 04829b4 to eead0c3 Compare January 24, 2024 06:06
@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Jan 24, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from eead0c3 to 91fd2db Compare January 24, 2024 06:06
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

@q10 q10 force-pushed the export-D53033916 branch from 91fd2db to 14aed1c Compare February 1, 2024 19:30
q10 added a commit to q10/FBGEMM that referenced this pull request Feb 1, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

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

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Feb 1, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 14aed1c to 3613a50 Compare February 1, 2024 19:31
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Feb 1, 2024
Summary:
Pull Request resolved: pytorch#2282

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916

fbshipit-source-id: 3611011ac93198d1b7a4b479f3af3322eb19e0f2
@q10 q10 force-pushed the export-D53033916 branch from 3613a50 to 9dd47cc Compare February 1, 2024 19:36
q10 added a commit to q10/FBGEMM that referenced this pull request Feb 5, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 9dd47cc to 772a04d Compare February 5, 2024 21:04
@facebook-github-bot
Copy link
Contributor

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

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

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 772a04d to d4fc3bb Compare February 5, 2024 21:04
@facebook-github-bot
Copy link
Contributor

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

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

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from d4fc3bb to 7bbb442 Compare February 5, 2024 21:05
@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Feb 6, 2024
Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Reviewed By: spcyppt

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 7bbb442 to 795c0a0 Compare February 6, 2024 18:53
@facebook-github-bot
Copy link
Contributor

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

Summary:

This follows up the work on D51865590 and D52679387 by plumbing the `uvm_cache_stats` argument passing up to the Python API level.  `local_uvm_cache_stats` is now zeroed out before the prefetch step as opposed to after, to allow for the data to be passed into the forward step.

This is a re-attempt of landing D51995949 with additions copied from D52670550

Reviewed By: spcyppt

Differential Revision: D53033916
@q10 q10 force-pushed the export-D53033916 branch from 795c0a0 to 1299965 Compare February 6, 2024 18:54
@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in c8cb1a2.

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