[pytorch] add missing host-device attribute to fix clang build (#37358)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37358
Test Plan:
```lang=bash
buck build mode/opt -c fbcode.cuda_use_clang=true //vision/fair/detectron2/tools:benchmark
```
Reviewed By: ngimel
Differential Revision: D21262235
fbshipit-source-id: 00633352d87da0881b2cc90759265fa0d0bd96be
