Bump to [email protected]

Project.toml

@@ -1,7 +1,7 @@
 name = "Checkpointing"
 uuid = "eb46d486-4f9c-4c3d-b445-a617f2a2f1ca"
 authors = ["Michel Schanen <[email protected]>", "Sri Hari Krishna Narayanan <[email protected]>"]
-version = "0.9.5"
+version = "0.9.6"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
@@ -14,7 +14,7 @@ Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 [compat]
 ChainRulesCore = "1"
 DataStructures = "0.18"
-Enzyme = "0.11, 0.12"
+Enzyme = "0.11, 0.12, 0.13"
 HDF5 = "0.16, 0.17"
 julia = "1.8"
 

src/Rules/EnzymeRules.jl

@@ -21,7 +21,7 @@ function augmented_primal(
 end
 
 function reverse(
-    config::ConfigWidth{1},
+    config,
     ::Const{typeof(Checkpointing.checkpoint_struct_for)},
     dret::Type{<:Const},
     tape,
@@ -60,7 +60,7 @@ function augmented_primal(
 end
 
 function reverse(
-    config::ConfigWidth{1},
+    config,
     ::Const{typeof(Checkpointing.checkpoint_struct_while)},
     dret::Type{<:Const},
     tape,