xla
parallel_loader: Fix TPU memory leak when calling iter.
#8039

Merged

parallel_loader: Fix TPU memory leak when calling iter. #8039

will-cromar merged 1 commit into pytorch:master from dudulightricks:bug-fix/mp-device-loader-memory-leak

JackCaoG requested a review from

will-cromar 1 year ago

will-cromar approved these changes on 2024-09-19

parallel_loader: Fix TPU memory leak when calling __iter__.

744bf9b3

dudulightricks force pushed from 984af380 to 744bf9b3 1 year ago

will-cromar merged d79a37c4 into master 1 year ago

Reviewers

will-cromar

Assignees

No one assigned

Labels

None yet

Milestone

No milestone