Skip to content
This repository has been archived by the owner on Feb 7, 2023. It is now read-only.

Commit

Permalink
s/CopyGPUToGPU/Copy
Browse files Browse the repository at this point in the history
Summary: CopyGPUToGPU does not exist. Copy seems to do the trick. Didn't go into details of how copy works, not sure if it ends up triggering UVA.

Reviewed By: akyrola

Differential Revision: D5471014

fbshipit-source-id: d8bc1aed9b19070c92f3ffc76f5617bdd0054563
  • Loading branch information
aazzolini authored and facebook-github-bot committed Jul 21, 2017
1 parent 3a0ad3f commit f5bbac6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion caffe2/python/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -2071,7 +2071,7 @@ def copy_func_between_devices(src, dst):
else:
def fun(net, *args, **kw):
with DeviceScope(dst):
return net.CopyGPUToGPU(*args, **kw)
return net.Copy(*args, **kw)
return fun

if src.device_type == CUDA and dst.device_type == CPU:
Expand Down

0 comments on commit f5bbac6

Please sign in to comment.