AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use HIP math functions, stop using hipLaunchParm.

Browse Source
This commit is contained in:
Alex Voicu
2018-10-10 11:56:54 +01:00
parent 4bc40551b5
commit 069bbbd7ed
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/src/kernel/hipLanguageExtensions.cpp
+5 -3
View File
@@ -28,6 +28,8 @@ THE SOFTWARE.
*/ */
#include "hip/hip_runtime.h" #include "hip/hip_runtime.h"
#include "hip/math_functions.h"
#include <test_common.h> #include <test_common.h>
#ifdef __HCC__ #ifdef __HCC__
@@ -53,7 +55,7 @@ __device__ __forceinline__ int sum1_forceinline(int a) { return a + 1; };
__device__ __host__ float PlusOne(float x) { return x + 1.0; } __device__ __host__ float PlusOne(float x) { return x + 1.0; }
__global__ void MyKernel(const hipLaunchParm lp, const float* a, const float* b, float* c, __global__ void MyKernel(const float* a, const float* b, float* c,
unsigned N) { unsigned N) {
// KERNELBEGIN; // KERNELBEGIN;
@@ -71,12 +73,12 @@ void callMyKernel() {
const unsigned blockSize = 256; const unsigned blockSize = 256;
unsigned N = blockSize; unsigned N = blockSize;
hipLaunchKernel(MyKernel, dim3(N / blockSize), dim3(blockSize), 0, 0, a, b, c, N); hipLaunchKernelGGL(MyKernel, dim3(N / blockSize), dim3(blockSize), 0, 0, a, b, c, N);
} }
template <typename T> template <typename T>
__global__ void vectorADD(const hipLaunchParm lp, T __restrict__* A_d, T* B_d, T* C_d, size_t N) { __global__ void vectorADD(T __restrict__* A_d, T* B_d, T* C_d, size_t N) {
// KERNELBEGIN; // KERNELBEGIN;
#ifdef NOT_YET #ifdef NOT_YET
int a = __shfl_up(x, 1); int a = __shfl_up(x, 1);