[ROCm] Add debug flag (#36521)
Summary:
This kernel debug flag should help locate the issues we are observing on
some of the CI nodes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36521
Differential Revision: D21010612
Pulled By: ezyang
fbshipit-source-id: d746e4eb0af832e770d2231bfee4154b6e703c19