pytorch

15949fc2 - [ROCm] Enable few test_prim UTs for ROCm (#88983)

[ROCm] Enable few test_prim UTs for ROCm (#88983) Pull Request resolved: https://github.com/pytorch/pytorch/pull/88983 Approved by: https://github.com/IvanYashchuk, https://github.com/jeffdaily, https://github.com/malfet