Skip to content

Commit 4dbb406

Browse files
jumerckxavik-pal
andcommitted
revert wrong formatting changes to Enzyme.jl
Co-authored-by: avik Co-authored-by: Avik Pal <avikpal@mit.edu>
1 parent 94ecaff commit 4dbb406

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

src/Enzyme.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ end
9696
@register_make_zero_inplace(Enzyme.remake_zero!)
9797

9898
function Enzyme.make_zero(
99-
::Type{RT}, seen::IdDict, prev::RT, (::Val{copy_if_inactive})=Val(false)
99+
::Type{RT}, seen::IdDict, prev::RT, ::Val{copy_if_inactive}=Val(false)
100100
)::RT where {copy_if_inactive,RT<:Union{RArray,RNumber}}
101101
if haskey(seen, prev)
102102
return seen[prev]
@@ -142,7 +142,7 @@ function EnzymeRules.augmented_primal(
142142
) where {RT}
143143
primargs = ntuple(Val(length(args))) do i
144144
Base.@_inline_meta
145-
return args[i].val
145+
args[i].val
146146
end
147147

148148
primal = if EnzymeCore.needs_primal(config)
@@ -162,7 +162,7 @@ function EnzymeRules.augmented_primal(
162162
else
163163
ntuple(Val(EnzymeRules.width(config))) do i
164164
Base.@_inline_meta
165-
return ConcretePJRTArray(
165+
ConcretePJRTArray(
166166
zeros(T.val, primargs...);
167167
client=XLA.client(uval.val),
168168
device=XLA.device(uval.val),
@@ -192,7 +192,7 @@ function EnzymeRules.reverse(
192192
) where {RT,N}
193193
ntuple(Val(N + 2)) do i
194194
Base.@_inline_meta
195-
return nothing
195+
nothing
196196
end
197197
end
198198

@@ -426,7 +426,7 @@ function overload_autodiff(
426426
else
427427
ntuple(Val(width)) do i
428428
Base.@_inline_meta
429-
return deepcopy(result)
429+
deepcopy(result)
430430
end
431431
end
432432
else

0 commit comments

Comments
 (0)