From 4b73eaaa78aa2142cbaad3002e9395f261b96159 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Sun, 28 Jul 2024 17:25:24 -0700
Subject: [PATCH] chore: bump version

---
 CHANGELOG.md    | 6 ++++++
 Project.toml    | 2 +-
 src/autodiff.jl | 4 ++--
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index c769a5f..f5312dc 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,6 +5,12 @@ All notable changes to this project since the release of v1 will be documented i
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.1.3] - 2024-08-08
+
+### Fixed
+
+  - Fixed non-public API usage of `AutoEnzyme`. [\[#28\]](https://github.com/LuxDL/LuxTestUtils.jl/pull/26)
+
 ## [1.1.2] - 2024-07-28
 
 ### Fixed
diff --git a/Project.toml b/Project.toml
index 337efe4..6411c06 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "LuxTestUtils"
 uuid = "ac9de150-d08f-4546-94fb-7472b5760531"
 authors = ["Avik Pal <avikpal@mit.edu>"]
-version = "1.1.2"
+version = "1.1.3"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
diff --git a/src/autodiff.jl b/src/autodiff.jl
index cdf3c71..1221ed7 100644
--- a/src/autodiff.jl
+++ b/src/autodiff.jl
@@ -11,7 +11,7 @@ end
 
 # Enzyme.jl
 function gradient(f::F, ::AutoEnzyme{Nothing}, args...) where {F}
-    return gradient(f, AutoEnzyme(Enzyme.Reverse), args...)
+    return gradient(f, AutoEnzyme(; mode=Enzyme.Reverse), args...)
 end
 
 function gradient(f::F, ad::AutoEnzyme{<:Enzyme.ReverseMode}, args...) where {F}
@@ -22,7 +22,7 @@ function gradient(f::F, ad::AutoEnzyme{<:Enzyme.ReverseMode}, args...) where {F}
         needs_gradient(x) && return Enzyme.Duplicated(x, Enzyme.make_zero(x))
         return Enzyme.Const(x)
     end
-    Enzyme.autodiff(ad.mode, f, Enzyme.Active, args_activity...)
+    Enzyme.autodiff(ad.mode, Enzyme.Const(f), Enzyme.Active, args_activity...)
     return Tuple(map(enumerate(args)) do (i, x)
         needs_gradient(x) && return args_activity[i].dval
         return CRC.NoTangent()