Commit Graph

5 Commits (780140599f7b4621bd29066a8ced1be120862a89)

Author SHA1 Message Date
guofei 9bea834ed4
Refine the unittest to support py38 (#27208)
5 years ago
lilong12 29de0d97a5
add the support to specify device index for device_guard (#24555)
5 years ago
Zhang Ting 3f371db82a
add op_device attr for backward op_desc, test=develop (#23062)
5 years ago
Zhang Ting eec10aaba2
set op_device for loss_op_desc (#23027)
5 years ago
Zhang Ting 4e8bc02461
add fluid.device_guard to specify the device type for Op (#22254)
5 years ago