[ROCm]Fixed ut test_memory_timeline (#96752)
Fixed test_memory_profiler::TestMemoryProfilerE2E::test_memory_timeline by changing the (arbitrary) threshold for logging. We observe differently-sized allocations on different AMD GPUs, so chose a higher threshold of 512 to account for those differences and yet satisfy the test requirements.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96752
Approved by: https://github.com/jithunnair-amd, https://github.com/kit1980
