From ae672297dc819028cd99904acda738f3d745f2f7 Mon Sep 17 00:00:00 2001
From: Aditya Atluri <adityaavinash1@gmail.com>
Date: Sat, 16 Apr 2016 17:38:06 -0500
Subject: [PATCH] added memcpy and p2p test

[ROCm/hip commit: 50d6cc6b115fdb521b01db98e7bbe594b4cec1f4]
---
 projects/hip/tests/src/CMakeLists.txt   |   3 +-
 projects/hip/tests/src/hipMemcpyAll.cpp | 128 ++++++++++++++++++++++++
 2 files changed, 130 insertions(+), 1 deletion(-)
 create mode 100644 projects/hip/tests/src/hipMemcpyAll.cpp

diff --git a/projects/hip/tests/src/CMakeLists.txt b/projects/hip/tests/src/CMakeLists.txt
index 5b1829286d..e58298248b 100644
--- a/projects/hip/tests/src/CMakeLists.txt
+++ b/projects/hip/tests/src/CMakeLists.txt
@@ -176,7 +176,7 @@ make_hip_executable (hipFuncGetDevice hipFuncGetDevice.cpp)
 make_hip_executable (hipFuncSetDevice hipFuncSetDevice.cpp)
 make_hip_executable (hipFuncDeviceSynchronize hipFuncDeviceSynchronize.cpp)
 make_hip_executable (hipPeerToPeer_simple hipPeerToPeer_simple.cpp) 
-
+make_hip_executable (hipMemcpyAll hipMemcpyAll.cpp)
 make_hip_executable (hipMultiThreadDevice hipMultiThreadDevice.cpp) 
 
 make_test(hip_ballot " " )
@@ -193,6 +193,7 @@ make_test(hipMemset --N 10013 --memsetval 0x5a )  # oddball size.
 make_test(hipMemset --N 256M  --memsetval 0xa6 )  # big copy
 make_test(hipGridLaunch " " )
 make_test(hipEnvVarDriver " " )
+make_test(hipMemcpyAll " ")
 #TODO -reenable
 #make_test(hipPointerAttrib " " )
 #make_test(hipMultiThreadStreams1 " " )  Fails if 0x3 specified, passes otherwise.  
diff --git a/projects/hip/tests/src/hipMemcpyAll.cpp b/projects/hip/tests/src/hipMemcpyAll.cpp
new file mode 100644
index 0000000000..c8484aa32f
--- /dev/null
+++ b/projects/hip/tests/src/hipMemcpyAll.cpp
@@ -0,0 +1,128 @@
+/*
+Copyright (c) 2015-2016 Advanced Micro Devices, Inc. All rights reserved.
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+
+#include<hip_runtime.h>
+#include<iostream>
+#include<assert.h>
+#include"test_common.h"
+
+#define len 1024*1024
+#define size len * sizeof(float)
+
+template<typename T>
+void hmemset(T *ptr, T value)
+{
+	for(int i=0;i<len;i++){
+		ptr[i] = value;
+	}
+}
+
+int main(){
+
+int num;
+hipGetDeviceCount(&num);
+if(num < 2)
+{
+    return 0;
+}
+
+float *h0, *h1;
+float *ph0, *ph1;
+float *d0, *d1;
+h0 = new float[len];
+h1 = new float[len];
+hmemset(h0, 1.0f);
+int gpu0 = 0, gpu1 = 1;
+hipSetDevice(gpu0);
+hipHostMalloc((void**)&ph0, size);
+hipMalloc(&d0, size);
+hipSetDevice(gpu1);
+hipHostMalloc((void**)&ph1, size);
+hipMalloc(&d1, size);
+hipSetDevice(gpu0);
+
+
+
+hipMemcpy(h1, h0, size, hipMemcpyDefault);
+hipMemcpy(ph0, h1, size, hipMemcpyDefault);
+hipMemcpy(ph1, ph0, size, hipMemcpyDefault);
+assert(h0[0] == ph1[0]);
+hmemset(ph1, 0.0f);
+hipMemcpy(h0, ph1, size, hipMemcpyDefault);
+assert(h0[0] == 0.0f);
+
+
+
+
+hipSetDevice(gpu0);
+hmemset(ph0, 2.0f);
+hipMemcpy(d0, ph0, size, hipMemcpyDefault);
+hipMemcpy(h0, d0, size, hipMemcpyDefault);
+
+assert(h0[0] == ph0[0]);
+hmemset(h0, 3.0f);
+hipMemcpy(d0, h0, size, hipMemcpyDefault);
+
+hipMemcpy(ph0, d0, size, hipMemcpyDefault);
+
+assert(h0[0] == ph0[0]);
+
+hipSetDevice(gpu1);
+hmemset(ph1, 2.0f);
+hipMemcpy(d1, ph1, size, hipMemcpyDefault);
+
+hipMemcpy(h1, d1, size, hipMemcpyDefault);
+
+assert(h1[0] == ph1[0]);
+hmemset(h1, 3.0f);
+hipMemcpy(d1, h1, size, hipMemcpyDefault);
+
+hipMemcpy(ph1, d1, size, hipMemcpyDefault);
+
+assert(h1[0] == ph1[0]);
+
+hipSetDevice(gpu0);
+hmemset(ph0, 4.0f);
+hipMemcpy(d0, ph0, size, hipMemcpyDefault);
+
+hipMemcpy(ph0, d0, size, hipMemcpyDefault);
+
+hipMemcpy(h0, d0, size, hipMemcpyDefault);
+
+assert(ph0[0] == 4.0f);
+assert(h0[0] == 4.0f);
+
+hipSetDevice(gpu1);
+hmemset(ph1, 5.0f);
+hipMemcpy(d1, ph1, size, hipMemcpyDefault);
+
+hipMemcpy(ph1, d1, size, hipMemcpyDefault);
+
+hipMemcpy(h1, d1, size, hipMemcpyDefault);
+
+assert(ph1[0] == 5.0f);
+assert(h1[0] == 5.0f);
+
+hipSetDevice(gpu0);
+hipMemcpy(d0, ph1, size, hipMemcpyDefault);
+
+hipMemcpy(d1, d0, size, hipMemcpyDefault);
+passed();
+}