Check number of threads before launching kernel (#174)

JuliaGPU · Nov 15, 2023 · a7a802c · a7a802c
1 parent 912fa05
commit a7a802c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/matmul.jl b/src/matmul.jl
@@ -36,6 +36,9 @@ function matmul(conf::Config, a, b, c, d;
 
     hostkernel = @cuda launch=false kernel(args...)
     attributes(hostkernel.fun)[CUDA.FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES] = shmem
+
+    threads ≤ CUDA.maxthreads(hostkernel) || throw(ConfigError("Requested too many threads for this kernel: This kernel can be launched using at most $(CUDA.maxthreads(hostkernel)) threads, while this configuration required $(threads)"))
+
     hostkernel(args...; threads, blocks, shmem)
 end