pytorch
pytorch copied to clipboard
Fix hipblasLT workaround amax bug
Pushing to our internal fork. Already merged upstream here: https://github.com/pytorch/pytorch/pull/123275
I need to do a rebase
@alugorey , Can please work on updating this PR.
This code was brought in as a part of these 2 commits https://github.com/ROCm/pytorch/commit/69c6e0b8516933d08ad54bb2756124d98300cc5b https://github.com/ROCm/pytorch/commit/2cb3301f807e2d58d719bbd976b68a13999f1c70 They have already been merged into rocm6.2_internal_testing. We can abandon this PR