From ea3ebcb3decbb34447f25302d9c9fd5413cbc88a Mon Sep 17 00:00:00 2001
From: Aditya Atluri <adityaavinash1@gmail.com>
Date: Thu, 31 Mar 2016 02:14:12 -0500
Subject: [PATCH] Added directory for stress testing

---
 tests/src/hipMemcpyAsync2.cpp        | 15 ++++++++
 tests/src/stress/README.md           |  2 +
 tests/src/stress/hipStressMemcpy.cpp | 56 ++++++++++++++++++++++++++++
 3 files changed, 73 insertions(+)
 create mode 100644 tests/src/hipMemcpyAsync2.cpp
 create mode 100644 tests/src/stress/README.md
 create mode 100644 tests/src/stress/hipStressMemcpy.cpp

diff --git a/tests/src/hipMemcpyAsync2.cpp b/tests/src/hipMemcpyAsync2.cpp
new file mode 100644
index 0000000000..4d7af21c72
--- /dev/null
+++ b/tests/src/hipMemcpyAsync2.cpp
@@ -0,0 +1,15 @@
+#include"test_common.h"
+
+#define SIZE 1024*1024
+
+int main(){
+    float *A, *Ad;
+    HIPCHECK(hipHostMalloc((void**)&A,SIZE, hipHostMallocDefault));
+    HIPCHECK(hipMalloc((void**)&Ad, SIZE));
+    hipStream_t stream;
+    HIPCHECK(hipStreamCreate(&stream));
+    for(int i=0;i<SIZE;i++){
+    HIPCHECK(hipMemcpyAsync(Ad, A, SIZE, hipMemcpyHostToDevice, stream));
+    HIPCHECK(hipDeviceSynchronize());
+    }
+}
diff --git a/tests/src/stress/README.md b/tests/src/stress/README.md
new file mode 100644
index 0000000000..fc8afedd21
--- /dev/null
+++ b/tests/src/stress/README.md
@@ -0,0 +1,2 @@
+## Stress Tests
+This directory consists of stress tests for HIP
diff --git a/tests/src/stress/hipStressMemcpy.cpp b/tests/src/stress/hipStressMemcpy.cpp
new file mode 100644
index 0000000000..3473246470
--- /dev/null
+++ b/tests/src/stress/hipStressMemcpy.cpp
@@ -0,0 +1,56 @@
+/*
+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<time.h>
+
+#define NUM_SIZE 8
+#define NUM_ITER 1 << 30
+static size_t size[NUM_SIZE];
+
+void setup(){
+	for(int i=0;i<NUM_SIZE;i++){
+		size[i] = 1<<(i+6); // start at 8 bytes
+	}
+}
+
+void valSet(int *A, int val, size_t size){
+	size_t len = size/sizeof(int);
+	for(int i=0;i<len;i++){
+		A[i] = val;
+	}
+}
+
+int main(){
+	setup();
+	int *A, *Ad;
+	for(int i=0;i<NUM_SIZE;i++){
+		A = (int*)malloc(size[i]);
+		valSet(A, 1, size[i]);
+		hipMalloc(&Ad, size[i]);
+		std::cout<<"Malloc success at size: "<<size[i]<<std::endl;
+		for(int j=0;j<NUM_ITER;j++){
+            std::cout<<"\r"<<"Iter: "<<j;
+			hipMemcpy(Ad, A, size[i], hipMemcpyHostToDevice);
+		}
+        std::cout<<std::endl;
+		hipDeviceSynchronize();
+	}
+}