Use thrust::host_vector instead of std::vector (#38178)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/38024.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38178
Differential Revision: D21502379
Pulled By: ezyang
fbshipit-source-id: 74dd6504c56f4150ed4cef129fd3f32f378c0564