File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 9696@register_make_zero_inplace (Enzyme. remake_zero!)
9797
9898function 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
197197end
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
You can’t perform that action at this time.
0 commit comments