AMD oom in `should_reduce_batch_size`
#812

Merged

Catch ROCM/HIP/AMD oom in `should_reduce_batch_size` #812

clefourrier merged 3 commits into huggingface:main from mcleish7:rocm-oom

Catch ROCM/HIP oom in should_reduce_batch_size

8f788a78

clefourrier approved these changes on 2025-06-20

Merge branch 'main' into rocm-oom

39f38eaa

fix formatting

5d2bfdba

clefourrier merged 95ec6691 into main 1 year ago

NathanHB added bug

Reviewers

clefourrier

Assignees

No one assigned

Labels

bug

Milestone

No milestone