xla
[Backport] Move CUDA tensors to CPU before moving to XLA
#6066
Merged
Overview
Commits
1
Changes
View On GitHub

[Backport] Move CUDA tensors to CPU before moving to XLA #6066

ManfeiBai merged 1 commit into r2.2 from wcromar/r2.2-fix-cuda-transfer
will-cromar
ysiraichi Move CUDA tensors to CPU before moving to XLA. (#6060)
653395bb
will-cromar will-cromar requested a review from zpcore zpcore 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
will-cromar will-cromar changed the title Move CUDA tensors to CPU before moving to XLA [Backport] Move CUDA tensors to CPU before moving to XLA 2 years ago
JackCaoG
JackCaoG approved these changes on 2023-12-08
ManfeiBai
ManfeiBai approved these changes on 2023-12-08
ManfeiBai ManfeiBai merged f77d1fac into r2.2 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum