We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a8f5ba7 commit 2ac439cCopy full SHA for 2ac439c
src/ATen/native/xpu/Copy.cpp
@@ -71,7 +71,12 @@ void memcpyAsync(
71
Device dst_device = iter.device(0);
72
Device src_device = iter.device(1);
73
if (dst_device == src_device) {
74
- copy_kernel(iter);
+ // copy_kernel(iter);
75
+ auto dst = (char*)iter.data_ptr(0);
76
+ auto src = (char*)iter.data_ptr(1);
77
+ size_t size = iter.numel() * iter.element_size(0);
78
+ auto q = copy_stream.queue();
79
+ q.copy(src, dst, size);
80
} else {
81
TORCH_INTERNAL_ASSERT(p2p_enabled == true);
82
auto dst = (char*)iter.data_ptr(0);
0 commit comments