From 3b5409c59a50297df8b3d1e6ca08453f3f424de1 Mon Sep 17 00:00:00 2001 From: scchan Date: Tue, 4 Oct 2016 15:36:44 -0500 Subject: [PATCH] don't use the copy api since it doesn't work with multi-gpu --- include/kalmar_runtime.h | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/include/kalmar_runtime.h b/include/kalmar_runtime.h index ac353553309..628c46f4ad0 100644 --- a/include/kalmar_runtime.h +++ b/include/kalmar_runtime.h @@ -528,12 +528,11 @@ static inline void copy_helper(std::shared_ptr& srcQueue, void* src /// If device pointer comes from cpu, let the device queue to handle the copy /// For example, if src is on cpu and dst is on device, /// in OpenCL, clEnqueueWrtieBuffer to write data from src to device - if (is_cpu_queue(srcQueue)) - dstQueue->write(dst, (char*)src + src_offset, cnt, dst_offset, block); - else if (is_cpu_queue(dstQueue)) + + if (is_cpu_queue(dstQueue)) srcQueue->read(src, (char*)dst + dst_offset, cnt, src_offset); else - dstQueue->copy(src, dst, cnt, src_offset, dst_offset, block); + dstQueue->write(dst, (char*)src + src_offset, cnt, dst_offset, block); } /// software MSI protocol