Junjie Bai e5fe66d7ea Add support for specifying device_option in Functional (#9619) ...

Summary:
e.g.
```
Functional.Add(x, y, device_option=DeviceOption(HIP, 0))

```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9619

Differential Revision: D8966599

Pulled By: bddppq

fbshipit-source-id: 22235e42f19278e79802642798bf0ee70a1202f6

2018-07-24 14:41:59 -07:00

4.2 KiB

Raw Blame History

View Raw