HABANA Device registration key and Autograd key addition (#57094)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57094
Reviewed By: mruberry
Differential Revision: D28355895
Pulled By: wconstab
fbshipit-source-id: 5d8b5762a69f444f4fe7f476891150fa5483d893
