comment gpu related stuff in eg to pass ci

Ferrite-FEM · Nov 19, 2024 · 6109bd1 · 6109bd1
1 parent 825d257
commit 6109bd1
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/docs/src/literate-tutorials/gpu_qp_heat_equation.jl b/docs/src/literate-tutorials/gpu_qp_heat_equation.jl
@@ -143,19 +143,19 @@ f = zeros(ndofs(dh));
 
 # Allocate GPU matrix
 ## commented to pass the test
-Kgpu = CUSPARSE.CuSparseMatrixCSC(K);
-fgpu = CUDA.zeros(ndofs(dh));
+## Kgpu = CUSPARSE.CuSparseMatrixCSC(K);
+## fgpu = CUDA.zeros(ndofs(dh));
 
 n_cells = dh |> get_grid |> getncells
 
 # Kernel configuration
 ## GPU kernel ##
 ## commented to pass the test
 ## First init the kernel with the required config.
-gpu_kernel = init_kernel(BackendCUDA, n_cells, n_basefuncs, assemble_gpu!, (Kgpu, fgpu, cellvalues, dh))
+## gpu_kernel = init_kernel(BackendCUDA, n_cells, n_basefuncs, assemble_gpu!, (Kgpu, fgpu, cellvalues, dh))
 ## Then launch the kernel
 ## gpu_kernel |> launch! or gpu_kernel()
-gpu_kernel()
+## gpu_kernel()
 
 ## CPU kernel ##
 ## cpu_kernel = init_kernel(BackendCPU, n_cells, n_basefuncs, assemble_gpu!, (K, f, cellvalues, dh));
@@ -171,5 +171,12 @@ norm(Kstd)
 
 
 ## GPU Benchmarking, remove when not needed ##
-## CUDA.@time gpu_kernel()
-## CUDA.@profile trace = true gpu_kernel()
+## function bench_gpu(n_cells, n_basefuncs, cellvalues, dh)
+##     Kgpu = CUSPARSE.CuSparseMatrixCSC(K);
+##     fgpu = CUDA.zeros(ndofs(dh));
+##     gpu_kernel = init_kernel(BackendCUDA, n_cells, n_basefuncs, assemble_gpu!, (Kgpu, fgpu, cellvalues, dh))
+##     gpu_kernel()
+## end
+
+## CUDA.@time bench_gpu(n_cells, n_basefuncs, cellvalues, dh)
+## CUDA.@profile trace = true bench_gpu(n_cells, n_basefuncs, cellvalues, dh)