minor fix

Ferrite-FEM · Dec 4, 2024 · 52f1479 · 52f1479
1 parent 441c9fb
commit 52f1479
Show file tree

Hide file tree

Showing 3 changed files with 3 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
@@ -171,8 +171,6 @@ gpu_kernel()
 ## cpu_kernel()
 
 
-Adapt.adapt_structure(CUSPARSE.CuSparseDeviceMatrixCSC, Kgpu)
-
 ## commented to pass the test
 ## norm(Kgpu)
 

diff --git a/ext/GPU/CUDAKernelLauncher.jl b/ext/GPU/CUDAKernelLauncher.jl
@@ -18,9 +18,9 @@ function Ferrite.init_kernel(::Type{BackendCUDA}, n_cells::Ti, n_basefuncs::Ti,
         threads = convert(Ti, min(n_cells, 256))
         shared_mem = _calculate_shared_memory(threads, n_basefuncs)
         blocks = _calculate_nblocks(threads, n_cells)
-        _adapted_args = _adapt_args(CuArray, args)
+        _adapted_args = _adapt_args(args)
 
-        if (_can_use_dynshmem(shared_mem) && false)
+        if (_can_use_dynshmem(shared_mem))
             Ke = DynamicSharedMemFunction{3, Float32, Int32}((threads, n_basefuncs, n_basefuncs), Int32(0))
             fe = DynamicSharedMemFunction{2, Float32, Int32}((threads, n_basefuncs), sizeof(Float32) * threads * n_basefuncs * n_basefuncs |> Int32)
             mem_alloc = SharedMemAlloc(Ke, fe, shared_mem)

diff --git a/ext/GPU/adapt.jl b/ext/GPU/adapt.jl
@@ -3,8 +3,7 @@
 Adapt.@adapt_structure GPUGrid
 Adapt.@adapt_structure GPUDofHandler
 
-function _adapt_args(to, args)
-    @show "Hi do you see me?"
+function _adapt_args(args)
     return tuple(((_adapt(arg) for arg in args) |> collect)...)
 end