search

ROCm / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
219 stars 50 forks source link

[release/1.13] Fix test_oom_tracing by increasing tensor size #1453

Closed dnikolaev-amd closed 1 week ago

dnikolaev-amd commented 1 week ago

test_oom_tracing expects OutOfMemory exception by allocating a large tensor. MI300X has enough memory to allocate test tensor This PR increases tensor size with a large margin to force OutOfMemory exception on MI300X and future GPU generations