Skip to content

Conversation

@botbw
Copy link
Contributor

@botbw botbw commented Oct 29, 2024

Add write_tensor, which moves tensors from GPU to CPU when needed:

For aio and uring backend, it does the transfer in blocking manner.

For pthread, it's non blocking and pipelined with file io:

image

@botbw botbw added the enhancement New feature or request label Oct 29, 2024
@botbw botbw requested a review from ver217 October 29, 2024 03:49
@botbw botbw self-assigned this Oct 29, 2024
@botbw botbw force-pushed the gpu2cpu branch 2 times, most recently from 8aa59d9 to 4d19a44 Compare October 29, 2024 08:49
@ver217 ver217 merged commit a1bf816 into main Nov 12, 2024
@ver217 ver217 deleted the gpu2cpu branch November 12, 2024 02:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants