Merge pull request shady-gang#36 from shady-gang/fix-checkerboard

make checkerboard sample work with CUDA

Author: Hugobros3

samples/checkerboard/checkerboard.c

@@ -73,15 +73,15 @@ int main(int argc, char **argv)
         error("Failed to load checkerboard module");
     Program* program = new_program_from_module(runtime, &compiler_config, m);
 
-    wait_completion(launch_kernel(program, device, "main", 16, 16, 1, 1, (void*[]) { &buf_addr }));
+    wait_completion(launch_kernel(program, device, "checkerboard", 16, 16, 1, 1, (void*[]) { &buf_addr }));
 
     copy_from_buffer(buf, 0, img, buf_size);
     info_print("data %d\n", (int) img[0]);
 
     destroy_buffer(buf);
 
     shutdown_runtime(runtime);
-    saveppm("ao.ppm", WIDTH, HEIGHT, img);
+    saveppm("checkerboard.ppm", WIDTH, HEIGHT, img);
     destroy_ir_arena(a);
     free(img);
 }

samples/checkerboard/checkerboard_kernel.slim

@@ -4,7 +4,7 @@ const i32 HEIGHT = 256;
 @Builtin("GlobalInvocationId")
 uniform input pack[u32; 3] global_id;
 
-@EntryPoint("Compute") @WorkgroupSize(16, 16, 1) fn main(uniform ptr global [u8] p) {
+@EntryPoint("Compute") @WorkgroupSize(16, 16, 1) fn checkerboard(uniform ptr global [u8] p) {
     val thread_id = global_id;
     val x = reinterpret[i32](thread_id#0);
     val y = reinterpret[i32](thread_id#1);