Skip to content

Commit 2ac439c

Browse files
committed
wa for d2d copy
1 parent a8f5ba7 commit 2ac439c

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/ATen/native/xpu/Copy.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,12 @@ void memcpyAsync(
7171
Device dst_device = iter.device(0);
7272
Device src_device = iter.device(1);
7373
if (dst_device == src_device) {
74-
copy_kernel(iter);
74+
// 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);
7580
} else {
7681
TORCH_INTERNAL_ASSERT(p2p_enabled == true);
7782
auto dst = (char*)iter.data_ptr(0);

0 commit comments

Comments
 (0)