Fix inactive rule to only mark constructor of one type as inactive, n…

…ot the constructor for _all_ types
JuliaGPU · Jul 23, 2023 · bb7ab80 · bb7ab80
1 parent 5d0365d
commit bb7ab80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/EnzymeExt.jl b/ext/EnzymeExt.jl
@@ -8,7 +8,7 @@ module EnzymeExt
     end
     import KernelAbstractions: Kernel, StaticSize, launch_config, __groupsize, __groupindex, blocks, mkcontext, CompilerMetadata, CPU
 
-    EnzymeRules.inactive(::typeof(StaticSize), x...) = nothing
+    EnzymeRules.inactive(::Type{StaticSize}, x...) = nothing
 
     function fwd(ctx, f, args...)
         EnzymeCore.autodiff_deferred(Forward, Const(f), Const, Const(ctx), args...)