From a352ee923e7cd42d00510b223761d4bb42b3e920 Mon Sep 17 00:00:00 2001 From: Ben Sander Date: Sun, 11 Sep 2016 10:29:18 -0500 Subject: [PATCH] Fix bugs for different CopyModes Change-Id: Ie59fb3b36a9ff2de178307a6d5756b5c71e0306b --- src/unpinned_copy_engine.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/unpinned_copy_engine.cpp b/src/unpinned_copy_engine.cpp index abeb5910e4..4ae6990180 100644 --- a/src/unpinned_copy_engine.cpp +++ b/src/unpinned_copy_engine.cpp @@ -189,6 +189,7 @@ void UnpinnedCopyEngine::CopyHostToDevice(UnpinnedCopyEngine::CopyMode copyMode, if (copyMode == UseMemcpy) { + CopyHostToDeviceMemcpy(dst, src, sizeBytes, waitFor); } else if (copyMode == UsePinInPlace) { @@ -316,7 +317,7 @@ void UnpinnedCopyEngine::CopyDeviceToHost(CopyMode copyMode ,void* dst, const vo if (copyMode == UsePinInPlace) { CopyDeviceToHostPinInPlace(dst, src, sizeBytes, waitFor); - } if (copyMode == UseStaging) { + } else if (copyMode == UseStaging) { CopyDeviceToHostStaging(dst, src, sizeBytes, waitFor); } else { // Unknown copy mode.