only runs mem leak check on master (#60023)
Summary:
setting environment variable to only do cuda mem leak check on master CI jobs.
See discussion in https://github.com/pytorch/pytorch/pull/59402#issuecomment-860773034
See stats before/after disabling mem leak check: https://github.com/pytorch/pytorch/pull/59942#issuecomment-860947095
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60023
Test Plan:
https://github.com/pytorch/pytorch/issues/60108
https://github.com/pytorch/pytorch/issues/60116
Reviewed By: janeyx99
Differential Revision: D29164182
Pulled By: walterddr
fbshipit-source-id: dfe88c2c1275b6eb35f18b58aacdc220f34ccb59
Author
Rong Rong (AI Infra)