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.

Sign up for GitHub

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

Jump to bottom

AIO CPU Locked Tensor #6592

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
Open

AIO CPU Locked Tensor #6592

wants to merge 11 commits into from

Conversation

jomayeri
Copy link
Contributor

Restoring the functionality of the cpu locked tensor in the AIO library.

@tjruwase tjruwase requested review from tjruwase and removed request for arashb and awan-10 October 1, 2024 13:50
@tjruwase
Copy link
Contributor

tjruwase commented Oct 1, 2024

@jomayeri, need to re-enable these unit tests

DeepSpeed/tests/unit/ops/aio/test_aio.py

Line 81 in 8cded57

@pytest.mark.parametrize("use_cuda_pinned_tensor", [True]) # TODO: aio_handle pinned tensor API is broken

@jomayeri jomayeri requested a review from loadams as a code owner October 1, 2024 18:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@tjruwase tjruwase tjruwase approved these changes

@loadams loadams Awaiting requested review from loadams loadams is a code owner

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@jomayeri @tjruwase