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

Test keeping output of a model on the GPU #273

Open
jwallwork23 opened this issue Jan 30, 2025 · 0 comments
Open

Test keeping output of a model on the GPU #273

jwallwork23 opened this issue Jan 30, 2025 · 0 comments
Labels
gpu Related to buiding and running on GPU testing Related to FTorch testing

Comments

@jwallwork23
Copy link
Contributor

The multi-GPU example currently puts the input tensors and model on CUDA devices, but has the output tensors on the CPU. We should test keeping the output on the GPU.
@jwallwork23 jwallwork23 added gpu Related to buiding and running on GPU testing Related to FTorch testing labels Jan 30, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
gpu Related to buiding and running on GPU testing Related to FTorch testing
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@jwallwork23