Update AdaptExt.jl

cc @vchuravy
EnzymeAD · Jun 21, 2024 · 884aed7 · 884aed7
1 parent 6b01be1
commit 884aed7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/lib/EnzymeCore/Project.toml b/lib/EnzymeCore/Project.toml
@@ -1,7 +1,7 @@
 name = "EnzymeCore"
 uuid = "f151be2c-9106-41f4-ab19-57ee4f262869"
 authors = ["William Moses <wmoses@mit.edu>", "Valentin Churavy <vchuravy@mit.edu>"]
-version = "0.7.5"
+version = "0.7.6"
 
 [compat]
 Adapt = "3, 4"

diff --git a/lib/EnzymeCore/ext/AdaptExt.jl b/lib/EnzymeCore/ext/AdaptExt.jl
@@ -15,5 +15,11 @@ end
 function Adapt.adapt_structure(to, x::BatchDuplicatedNoNeed)
     return BatchDuplicatedNoNeed(adapt(to, x.val), adapt(to, x.dval))
 end
+function Adapt.adapt_structure(to, x::MixedDuplicated)
+    return MixedDuplicated(adapt(to, x.val), adapt(to, x.dval))
+end
+function Adapt.adapt_structure(to, x::BatchMixedDuplicated)
+    return BatchMixedDuplicated(adapt(to, x.val), adapt(to, x.dval))
+end
 
 end #module
diff --git a/lib/EnzymeCore/src/EnzymeCore.jl b/lib/EnzymeCore/src/EnzymeCore.jl
@@ -3,6 +3,7 @@ module EnzymeCore
 export Forward, Reverse, ReverseWithPrimal, ReverseSplitNoPrimal, ReverseSplitWithPrimal
 export ReverseSplitModified, ReverseSplitWidth, ReverseHolomorphic, ReverseHolomorphicWithPrimal
 export Const, Active, Duplicated, DuplicatedNoNeed, BatchDuplicated, BatchDuplicatedNoNeed
+export MixedDuplicated, MixedDuplicatedNoNeed
 export DefaultABI, FFIABI, InlineABI
 export BatchDuplicatedFunc