From 271e2357e0a95a60e70478dd6d5d128edce8a03e Mon Sep 17 00:00:00 2001
From: Ben Sander <ben.sander@amd.com>
Date: Sun, 11 Sep 2016 10:29:18 -0500
Subject: [PATCH] Fix bugs for different CopyModes

Change-Id: Ie59fb3b36a9ff2de178307a6d5756b5c71e0306b
---
 hipamd/src/unpinned_copy_engine.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/unpinned_copy_engine.cpp b/hipamd/src/unpinned_copy_engine.cpp
index abeb5910e4..4ae6990180 100644
--- a/hipamd/src/unpinned_copy_engine.cpp
+++ b/hipamd/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.