diff --git a/hipamd/tests/src/hipDeviceBody.cpp b/hipamd/tests/src/hipDeviceBody.cpp
index fb0b980713..2e740cf748 100644
--- a/hipamd/tests/src/hipDeviceBody.cpp
+++ b/hipamd/tests/src/hipDeviceBody.cpp
@@ -1,5 +1,5 @@
 #include"hipDeviceHeader.h"
 
-__device__ int add(int &a, int &b){
+__host__ __device__ int add(int &a, int &b){
     return a+b;
 }
diff --git a/hipamd/tests/src/hipDeviceHeader.h b/hipamd/tests/src/hipDeviceHeader.h
index 265a2073c6..eb7bd05aa5 100644
--- a/hipamd/tests/src/hipDeviceHeader.h
+++ b/hipamd/tests/src/hipDeviceHeader.h
@@ -1,6 +1,7 @@
 #ifndef HIPDEVICEHEADER_H
 #define HIPDEVICEHEADER_H
 #include"test_common.h"
-__device__ int add(int a, int b);
+
+__host__ __device__ int add(int a, int b);
 
 #endif