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

Add/modify LXU cache lookup ops for pipeline prefetching #2154

Closed
wants to merge 1 commit into from

Conversation

sryap
Copy link
Contributor

@sryap sryap commented Nov 22, 2023

Summary:
This diff adds/updates LXU cache APIs for pipeline prefetching:

  • Update lxu_cache_lookup to allow for unique linear cache indices
    lookup and external output tensor to be passed to the op
  • Update lxu_cache_locations_update to support unique cache
    locations update
  • Add Python binding for get_unique_indices

Differential Revision: D51532548

Copy link

netlify bot commented Nov 22, 2023

Deploy Preview for pytorch-fbgemm-docs canceled.

Name Link
🔨 Latest commit 4da1f0c
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/6565148bcb9bb2000859e4b4

@facebook-github-bot
Copy link
Contributor

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 22, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 22, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

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

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

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

@sryap sryap force-pushed the export-D51532548 branch 2 times, most recently from 976d81e to 402f37f Compare November 27, 2023 20:18
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

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

sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 27, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

Differential Revision: D51532548
sryap pushed a commit to sryap/FBGEMM that referenced this pull request Nov 28, 2023
Summary:

This diff adds/updates LXU cache APIs for pipeline prefetching:

- Update `lxu_cache_lookup` to allow for unique linear cache indices
lookup and external output tensor to be passed to the op
- Update `lxu_cache_locations_update` to support unique cache
locations update
- Add Python binding for `get_unique_indices`

Reviewed By: levythu

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

This pull request has been merged in 18af2b2.

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.

None yet

2 participants