aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src/cuda-fractals.cu
diff options
context:
space:
mode:
Diffstat (limited to 'src/cuda-fractals.cu')
-rw-r--r--src/cuda-fractals.cu306
1 files changed, 306 insertions, 0 deletions
diff --git a/src/cuda-fractals.cu b/src/cuda-fractals.cu
new file mode 100644
index 0000000..92be53f
--- /dev/null
+++ b/src/cuda-fractals.cu
@@ -0,0 +1,306 @@
+#include <cuda_runtime.h>
+#include <thrust/complex.h>
+#include "fractals.h"
+
+//These won't work/ need to be modifiied for the cuda version
+//#include "grids.h"
+//#include "fractals.h"
+
+/*
+ * Macro for checking CUDA errors
+ */
+#define CHECK(call) \
+{ \
+ const cudaError_t error = call; \
+ if(error != cudaSuccess){ \
+ fprintf(stderr, "Error: %s:%d, ", __FILE__, __LINE__); \
+ fprintf(stderr, "Code: %d, Reason: %s\n", error, cudaGetErrorString(error)); \
+ } \
+}
+
+#ifndef CBASE
+#define CBASE double
+#endif
+
+#define BLOCK_SIZE_X 16
+#define BLOCK_SIZE_Y 16
+
+// this ain't pretty, but this logic should never change for kernels
+#define SET_ROW_COL \
+ const size_t row = blockIdx.y * blockDim.y + threadIdx.y; \
+ const size_t col = blockIdx.x * blockDim.x + threadIdx.x; \
+ if(row >= rows || col >= cols) return
+
+
+__device__
+thrust::complex<CBASE> grid_to_complex(const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t row, const size_t col, const size_t rows, const size_t cols){
+ const CBASE x_min = lower_left.real();
+ const CBASE x_max = upper_right.real();
+ const CBASE y_min = lower_left.imag();
+ const CBASE y_max = upper_right.imag();
+
+ const CBASE x_step = (x_max - x_min) / (CBASE)cols;
+ const CBASE y_step = (y_max - y_min) / (CBASE)rows;
+
+ const CBASE x = x_min + col * x_step;
+ const CBASE y = y_min + row * y_step;
+ const thrust::complex<CBASE> z(x,y);
+ return z;
+}
+
+__device__
+size_t mandelbrot(const thrust::complex<CBASE> z0, const size_t max_iterations){
+ thrust::complex<CBASE> z = z0;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= 2 && iteration < max_iterations){
+ z = z*z + z0;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void mandelbrot_kernel(size_t* grid_data, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = mandelbrot(z, max_iterations);
+}
+
+__device__
+size_t tricorn(const thrust::complex<CBASE> z0, const size_t max_iterations){
+ thrust::complex<CBASE> z = z0;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= 2 && iteration < max_iterations){
+ z = thrust::conj(z*z) + z0;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void tricorn_kernel(size_t* grid_data, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = tricorn(z, max_iterations);
+}
+
+__device__
+size_t burning_ship(const thrust::complex<CBASE> z0, const size_t max_iterations){
+ thrust::complex<CBASE> z = z0;
+ thrust::complex<CBASE> z_mod;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= 2 && iteration < max_iterations){
+ z_mod = thrust::complex<CBASE>(thrust::abs(z.real()), thrust::abs(z.imag()));
+ z = z_mod * zmod + z0;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void burning_ship_kernel(size_t* grid_data, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = burning_ship(z, max_iterations);
+}
+
+__device__
+size_t multibrot(const thrust::complex<CBASE> z0, const size_t max_iterations, const double d){
+ thrust::complex<CBASE> z = z0;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= 2 && iteration < max_iterations){
+ z = thrust::pow(z, d) + z0;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void multibrot_kernel(size_t* grid_data, const double degree, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = multibrot(z, max_iterations, degree);
+}
+
+__device__
+size_t multicorn(const thrust::complex<CBASE> z0, const size_t max_iterations, const double d){
+ thrust::complex<CBASE> z = z0;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= 2 && iteration < max_iterations){
+ z = thrust::conj(thrust::pow(z, d)) + z0;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void multicorn_kernel(size_t* grid_data, const double degree, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = multicorn(z, max_iterations, degree);
+}
+
+__device__
+size_t julia(const thrust::complex<CBASE> z0, const size_t max_iterations, const thrust::complex<CBASE> c, const double R){
+ thrust::complex<CBASE> z = z0;
+ size_t iteration = 0;
+ while(thrust::abs(z) <= R && iteration < max_iterations){
+ z = z*z + c;
+ iteration++;
+ }
+ return iteration;
+}
+
+__global__
+void julia_kernel(size_t* grid_data, const thrust:complex<CBASE> constant, const double radius, const size_t max_iterations, const thrust::complex<CBASE> lower_left, const thrust::complex<CBASE> upper_right, const size_t rows, const size_t cols){
+ SET_ROW_COL;
+
+ const auto z = grid_to_complex(lower_left, upper_right, row, col, rows, cols);
+
+ grid_data[row*cols + col] = julia(z, max_iterations, constant, radius);
+}
+
+extern "C" {
+void mandelbrot_grid(grid_t* grid, const grid_gen_params* params){
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ //TODO: find good sizes
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ //dim3 grid_size(0,0);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ mandelbrot_kernel<<<grid_size, block_size>>>(d_grid_data, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+
+void tricorn_grid(grid_t* grid, const grid_gen_params* params){
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ tricorn_kernel<<<grid_size, block_size>>>(d_grid_data, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+
+void burning_ship_grid(grid_t* grid, const grid_gen_params* params){
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ burning_ship_kernel<<<grid_size, block_size>>>(d_grid_data, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+
+void multibrot_grid(grid_t* grid, const grid_gen_params* params){
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ const double degree = params->degree;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ multibrot_kernel<<<grid_size, block_size>>>(d_grid_data, degree, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+
+void multicorn_grid(grid_t* grid, const grid_gen_params* params){
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ const double degree = params->degree;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ multicorn_kernel<<<grid_size, block_size>>>(d_grid_data, degree, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+
+void julia_grid(grid_t* grid, const grid_gen_params* params){
+ const complex_t constant = params->cr.constant;
+ const double radius = params->cr.radius;
+ const size_t size = grid->size;
+ const size_t rows = grid->y;
+ const size_t cols = grid->x;
+ const size_t max_iterations = grid->max_iterations;
+ const double degree = params->degree;
+ thrust::complex<CBASE> lower_left(grid->lower_left.re, grid->lower_left.im);
+ thrust::complex<CBASE> upper_right(grid->upper_right.re, grid->upper_right.im);
+
+ size_t* d_grid_data;
+ CHECK(cudaMalloc(&d_grid_data, size*sizeof(size_t)));
+ dim3 block_size(BLOCK_SIZE_X, BLOCK_SIZE_Y);
+ dim3 grid_size((cols + block_size.x - 1) / block_size.x, (rows + block_size.y - 1) / block_size.y);
+ multicorn_kernel<<<grid_size, block_size>>>(d_grid_data, constant, radius, max_iterations, lower_left, upper_right, rows, cols);
+ CHECK(cudaDeviceSynchronize());
+
+ CHECK(cudaMemcpy(grid->data, d_grid_data, size*sizeof(size_t), cudaMemcpyDeviceToHost));
+
+ CHECK(cudaFree(d_grid_data));
+ CHECK(cudaDeviceReset());
+}
+}
generated by cgit v1.2.3 (git 2.39.1) at 2025-12-06 10:11:55 +0000