Migrate THAllocator to ATen (#60325)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60325
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D29371715
Pulled By: ngimel
fbshipit-source-id: 78ec8368a48e1a4690d0664a0b02d2a235af98ff