Evgeny Mankov
368977f75b
+ Add API_COMPLEX support (data types and functions) + Add cuComplex_API_supported_by_HIP.md + Add cuComplex_Julia.cu test + Update README.md
59 línte
1.6 KiB
Plaintext
59 línte
1.6 KiB
Plaintext
// RUN: %run_test hipify "%s" "%t" %cuda_args
|
|
|
|
// CHECK: #include <hip/hip_runtime.h>
|
|
// CHECK: #include "hip/hip_complex.h"
|
|
#include "cuComplex.h"
|
|
|
|
#define TYPEFLOAT
|
|
#define DIMX 100
|
|
#define DIMY 40
|
|
#define moveX 2
|
|
#define moveY 1
|
|
|
|
#define MAXITERATIONS 10
|
|
|
|
#ifdef TYPEFLOAT
|
|
#define TYPE float
|
|
// CHECK: #define cTYPE hipFloatComplex
|
|
#define cTYPE cuFloatComplex
|
|
// CHECK: #define cMakecuComplex(re,i) make_hipFloatComplex(re,i)
|
|
#define cMakecuComplex(re,i) make_cuFloatComplex(re,i)
|
|
#endif
|
|
#ifdef TYPEDOUBLE
|
|
// CHECK: #define TYPE hipDoubleComplex
|
|
#define TYPE cuDoubleComplex
|
|
// CHECK: #define cMakecuComplex(re,i) make_hipDoubleComplex(re,i)
|
|
#define cMakecuComplex(re,i) make_cuDoubleComplex(re,i)
|
|
#endif
|
|
|
|
__device__ cTYPE juliaFunctor(cTYPE p, cTYPE c) {
|
|
// CHECK: return hipCaddf(hipCmulf(p, p), c);
|
|
return cuCaddf(cuCmulf(p, p), c);
|
|
}
|
|
|
|
__device__ cTYPE convertToComplex(int x, int y, float zoom) {
|
|
TYPE jx = 1.5 * (x - DIMX / 2) / (0.5 * zoom * DIMX) + moveX;
|
|
TYPE jy = (y - DIMY / 2) / (0.5 * zoom * DIMY) + moveY;
|
|
return cMakecuComplex(jx, jy);
|
|
}
|
|
|
|
__device__ int evolveComplexPoint(cTYPE p, cTYPE c) {
|
|
int it = 1;
|
|
// CHECK: while (it <= MAXITERATIONS && hipCabsf(p) <= 4) {
|
|
while (it <= MAXITERATIONS && cuCabsf(p) <= 4) {
|
|
p = juliaFunctor(p, c);
|
|
it++;
|
|
}
|
|
return it;
|
|
}
|
|
|
|
__global__ void computeJulia(int* data, cTYPE c, float zoom) {
|
|
int i = blockIdx.x * blockDim.x + threadIdx.x;
|
|
int j = blockIdx.y * blockDim.y + threadIdx.y;
|
|
|
|
if (i<DIMX && j<DIMY) {
|
|
cTYPE p = convertToComplex(i, j, zoom);
|
|
data[i*DIMY + j] = evolveComplexPoint(p, c);
|
|
}
|
|
}
|