Handle static arrays with Mooncake (#643)

JuliaDiff · Dec 6, 2024 · c20083f · c20083f
1 parent ae87c32
commit c20083f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...nInterface/ext/DifferentiationInterfaceMooncakeExt/DifferentiationInterfaceMooncakeExt.jl b/...nInterface/ext/DifferentiationInterfaceMooncakeExt/DifferentiationInterfaceMooncakeExt.jl
@@ -16,7 +16,7 @@ using Mooncake:
 DI.check_available(::AutoMooncake) = true
 
 copyto!!(dst::Number, src::Number) = convert(typeof(dst), src)
-copyto!!(dst, src) = copyto!(dst, src)
+copyto!!(dst, src) = DI.ismutable_array(dst) ? copyto!(dst, src) : convert(typeof(dst), src)
 
 get_config(::AutoMooncake{Nothing}) = Config()
 get_config(backend::AutoMooncake{<:Config}) = backend.config