ROCm / pytorch

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

Added a fraction parameter for profiler_oom_tracing #1411

Closed BLOrange-AMD closed 1 month ago

BLOrange-AMD commented 2 months ago

Fixes https://ontrack-internal.amd.com/browse/SWDEV-440517 profiler/test_profiler.py::TestProfiler::test_oom_tracing in release/2.0 branch. Use dynamic ratio calculation to support all gfx rather than MI3xx check only.