No commit message

paulxshen · Jan 13, 2025 · bf4826a · bf4826a
1 parent d4ea7a1
commit bf4826a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/del.jl b/src/del.jl
@@ -14,8 +14,8 @@ function diffpad(a, vl, vr=vl; dims=1, diff=diff, autodiff=true)
     # println()
 
     if autodiff
-        vl ∈ (0, nothing) && @nogradvars vl
-        vr ∈ (0, nothing) && @nogradvars vr
+        vl ∈ (0, nothing) && @nograd vl
+        vr ∈ (0, nothing) && @nograd vr
         b = diff(a; dims)
         b = pad(b, vl, vr, l * sel, r * sel)
     else