Shangdi Yu
|
2e440e39a6
|
[nativert] Move Placement to pytorch core (#152953)
Summary:
Move Placement to pytorch core.
Using `torch::nativert::isSameDevice` explicitly in code to avoid confusion with the `isSameDevice` in torch namespace.
Test Plan:
```
buck run fbcode//mode/dev-nosan //caffe2/test/cpp/nativert:placement_test
./bin/test_nativert
```
OSS and internal CI
Differential Revision: D74190745
Pull Request resolved: https://github.com/pytorch/pytorch/pull/152953
Approved by: https://github.com/Skylion007, https://github.com/swolchok, https://github.com/zhxchen17, https://github.com/cyyever
|
2025-05-14 15:26:54 +00:00 |
|