[ROCm] enable mem leak check for rocm (#35953)
Summary:
CC iotamudelta
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35953
Differential Revision: D21742926
Pulled By: zou3519
fbshipit-source-id: f18534dbb88a84fe98b8d85ce8fde652916a72d5