Fix adapt call

CliMA · Oct 23, 2024 · 71f1a38 · 71f1a38
1 parent 72cc110
commit 71f1a38
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ext/cuda/data_layouts.jl b/ext/cuda/data_layouts.jl
@@ -33,20 +33,20 @@ include("data_layouts_fused_copyto.jl")
 include("data_layouts_mapreduce.jl")
 include("data_layouts_threadblock.jl")
 
+adapt_f(to, f::F) where {F} = Adapt.adapt(to, f)
+adapt_f(to, ::Type{F}) where {F} = (x...) -> F(x...)
+
 function Adapt.adapt_structure(
     to::CUDA.KernelAdaptor,
     bc::DataLayouts.NonExtrudedBroadcasted{Style},
 ) where {Style}
     DataLayouts.NonExtrudedBroadcasted{Style}(
-        Adapt.adapt_f(to, bc.f),
+        adapt_f(to, bc.f),
         Adapt.adapt(to, bc.args),
         Adapt.adapt(to, bc.axes),
     )
 end
 
-adapt_f(to, f::F) where {F} = Adapt.adapt(to, f)
-adapt_f(to, ::Type{F}) where {F} = (x...) -> F(x...)
-
 function Adapt.adapt_structure(
     to::CUDA.KernelAdaptor,
     fmbc::FusedMultiBroadcast,