From bdbc4ce11cdaed2274fda67a1460da8a0bae8406 Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Tue, 24 Oct 2017 18:12:25 +0530
Subject: [PATCH] Example showing globals use with module APIs

[ROCm/hip commit: d638cd4fc177403f3208f4336e0736dc3fa9c61b]
---
 .../0_Intro/module_api_global/Makefile        |  17 ++
 .../0_Intro/module_api_global/runKernel.cpp   | 167 ++++++++++++++++++
 .../0_Intro/module_api_global/vcpy_kernel.cpp |  41 +++++
 3 files changed, 225 insertions(+)
 create mode 100644 projects/hip/samples/0_Intro/module_api_global/Makefile
 create mode 100644 projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
 create mode 100644 projects/hip/samples/0_Intro/module_api_global/vcpy_kernel.cpp

diff --git a/projects/hip/samples/0_Intro/module_api_global/Makefile b/projects/hip/samples/0_Intro/module_api_global/Makefile
new file mode 100644
index 0000000000..97605020b2
--- /dev/null
+++ b/projects/hip/samples/0_Intro/module_api_global/Makefile
@@ -0,0 +1,17 @@
+HIP_PATH?= $(wildcard /opt/rocm/hip)
+ifeq (,$(HIP_PATH))
+	HIP_PATH=../../..
+endif
+HIPCC=$(HIP_PATH)/bin/hipcc
+HIP_PLATFORM=$(shell $(HIP_PATH)/bin/hipconfig --compiler)
+
+all: vcpy_kernel.code runKernel.hip.out
+
+runKernel.hip.out: runKernel.cpp
+	$(HIPCC) $(HIPCC_FLAGS) $< -o $@
+
+vcpy_kernel.code: vcpy_kernel.cpp
+	$(HIPCC)  --genco  $(GENCO_FLAGS) $^ -o $@
+
+clean:
+	rm -f *.code *.out
diff --git a/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp b/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
new file mode 100644
index 0000000000..3f84720edf
--- /dev/null
+++ b/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
@@ -0,0 +1,167 @@
+/*
+Copyright (c) 2017 - present 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/hip_runtime.h"
+#include "hip/hip_runtime_api.h"
+#include <iostream>
+#include <fstream>
+#include <vector>
+#include <hip/hip_hcc.h>
+
+#define LEN 64
+#define SIZE LEN*sizeof(float)
+
+#define fileName "vcpy_kernel.code"
+float myDeviceGlobal;
+float myDeviceGlobalArray[16];
+#define HIP_CHECK(cmd) \
+{\
+    hipError_t status = cmd;\
+    if(status != hipSuccess) {std::cout<<"error: #"<<status<<" ("<< hipGetErrorString(status) << ") at line:"<<__LINE__<<":  "<<#cmd<<std::endl;abort();}\
+}
+
+int main(){
+    float *A, *B;
+    float* Ad, *Bd;
+    A = new float[LEN];
+    B = new float[LEN];
+
+    for(uint32_t i=0;i<LEN;i++){
+        A[i] = i*1.0f;
+        B[i] = 0.0f;
+    }
+
+    hipInit(0);
+    hipDevice_t device;
+    hipCtx_t context;
+    hipDeviceGet(&device, 0);
+    hipCtxCreate(&context, 0, device);
+
+    hipMalloc((void**)&Ad, SIZE);
+    hipMalloc((void**)&Bd, SIZE);
+
+    hipMemcpyHtoD(hipDeviceptr_t(Ad), A, SIZE);
+    hipMemcpyHtoD((hipDeviceptr_t)(Bd), B, SIZE);
+    hipModule_t Module;
+    HIP_CHECK(hipModuleLoad(&Module, fileName));
+
+    float myDeviceGlobal_h = 42.0;
+    myDeviceGlobal = 42.0;
+
+#define ARRAY_SIZE 16
+
+    float myDeviceGlobalArray_h[ARRAY_SIZE];
+    for (int i=0; i<ARRAY_SIZE; i++) {
+        myDeviceGlobalArray_h[i] = i*1000.0f;
+        myDeviceGlobalArray[i] = i*1000.0f;
+    }
+
+#ifdef __HIP_PLATFORM_HCC__
+		uint32_t len = LEN;
+		uint32_t one = 1;
+
+    struct {
+        uint32_t _hidden[6];  // genco path + wrapper-gen pass use hidden arguments.
+        void * _Ad;
+        void * _Bd;
+    } args;
+
+    args._Ad = Ad;
+    args._Bd = Bd;
+
+#endif
+
+#ifdef __HIP_PLATFORM_NVCC__
+    struct {
+        uint32_t _hidden[1];
+        void * _Ad;
+        void * _Bd;
+    } args;
+
+    args._hidden[0] = 0;
+    args._Ad = Ad;
+    args._Bd = Bd;
+#endif
+
+
+    size_t size = sizeof(args);
+
+    void *config[] = {
+      HIP_LAUNCH_PARAM_BUFFER_POINTER, &args,
+      HIP_LAUNCH_PARAM_BUFFER_SIZE, &size,
+      HIP_LAUNCH_PARAM_END
+    };
+
+    { 
+        hipFunction_t Function;
+        HIP_CHECK(hipModuleGetFunction(&Function, Module, "hello_world"));
+        HIP_CHECK(hipModuleLaunchKernel(Function, 1, 1, 1, LEN, 1, 1, 0, 0, NULL, (void**)&config));
+
+        hipMemcpyDtoH(B, Bd, SIZE);
+
+        int mismatchCount = 0;
+        for(uint32_t i=0;i<LEN;i++){
+            if (A[i] != B[i]) {
+                mismatchCount++;
+                std::cout<<"error: mismatch " << A[i]<<" != "<<B[i]<<std::endl;
+                if (mismatchCount >= 10) {
+                    break;
+                }
+            }
+        }
+
+        if (mismatchCount == 0) {
+            std::cout << "PASSED!\n";
+        } else {
+            std::cout << "FAILED!\n";
+        };
+    }
+
+    { 
+        hipFunction_t Function;
+        HIP_CHECK(hipModuleGetFunction(&Function, Module, "test_globals"));
+        HIP_CHECK(hipModuleLaunchKernel(Function, 1, 1, 1, LEN, 1, 1, 0, 0, NULL, (void**)&config));
+
+        hipMemcpyDtoH(B, Bd, SIZE);
+
+        int mismatchCount = 0;
+        for(uint32_t i=0;i<LEN;i++){
+            float expected = A[i] + myDeviceGlobal_h + myDeviceGlobalArray_h[i%16];
+            if (expected != B[i]) {
+                mismatchCount++;
+                std::cout<<"error: mismatch " << expected <<" != "<<B[i]<<std::endl;
+                if (mismatchCount >= 10) {
+                    break;
+                }
+            }
+        }
+
+        if (mismatchCount == 0) {
+            std::cout << "PASSED!\n";
+        } else {
+            std::cout << "FAILED!\n";
+        };
+    }
+
+    hipCtxDestroy(context);
+    return 0;
+}
diff --git a/projects/hip/samples/0_Intro/module_api_global/vcpy_kernel.cpp b/projects/hip/samples/0_Intro/module_api_global/vcpy_kernel.cpp
new file mode 100644
index 0000000000..f24a261585
--- /dev/null
+++ b/projects/hip/samples/0_Intro/module_api_global/vcpy_kernel.cpp
@@ -0,0 +1,41 @@
+/*
+Copyright (c) 2017 - present 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/hip_runtime.h"
+
+#define ARRAY_SIZE (16)
+
+extern float myDeviceGlobal;
+extern float myDeviceGlobalArray[16];;
+
+extern "C" __global__ void hello_world(hipLaunchParm lp, const float *a, float *b)
+{
+    int tx = hipThreadIdx_x;
+    b[tx] = a[tx];
+}
+
+extern "C" __global__ void test_globals(hipLaunchParm lp, const float *a, float *b)
+{
+    int tx = hipThreadIdx_x;
+    b[tx] = a[tx] + myDeviceGlobal+ myDeviceGlobalArray[tx%ARRAY_SIZE] ;
+}
+