Use gridDimHEADmain

1 files changed, 4 insertions, 4 deletions

diff --git a/julia/julia.cu b/julia/julia.cu
index f3ecb80..ed7b00a 100644
--- a/julia/julia.cu
+++ b/julia/julia.cu
@@ -39,7 +39,7 @@ __device__ int julia(int width, int height, int x, int y) {
   return 1;
 }
 
-__global__ void juliaMain(int width, int height, Pixel* image) {
+__global__ void juliaMain(Pixel* image) {
   const int x = blockIdx.x;
   const int y = blockIdx.y;
 
@@ -47,9 +47,9 @@ __global__ void juliaMain(int width, int height, Pixel* image) {
   constexpr Pixel juliaColour{228, 192, 135};
 
   const Pixel pixel =
-      julia(width, height, x, y) == 1 ? juliaColour : background;
+      julia(gridDim.x, gridDim.y, x, y) == 1 ? juliaColour : background;
 
-  image[y * width + x] = pixel;
+  image[y * gridDim.x + x] = pixel;
 }
 
 bool write_pbm(const Pixel* image, int width, int height, const char* path) {
@@ -85,7 +85,7 @@ int main(int argc, const char** argv) {
     goto cleanup;
   }
 
-  juliaMain<<<dim, 1>>>(width, height, image_dev);
+  juliaMain<<<dim, 1>>>(image_dev);
 
   if (cudaMemcpy(
           image_host, image_dev, image_size_bytes, cudaMemcpyDeviceToHost) !=