Apply suggestions from code review

Co-authored-by: Anton Smirnov <[email protected]>
FluxML · Jul 27, 2024 · 1568085 · 1568085
1 parent b3c7eb9
commit 1568085
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/FluxAMDGPUExt/functor.jl b/ext/FluxAMDGPUExt/functor.jl
@@ -81,8 +81,8 @@ function _amd(id::Union{Nothing, Int}, x)
     fmap(x -> Adapt.adapt(FluxAMDGPUAdaptor(id), x), x; exclude=_exclude)
 end
 
-_other_args(m::Conv) = [m.stride, m.pad, m.dilation, m.groups]
-_other_args(m::ConvTranspose) = [m.stride, m.pad, m.outpad, m.dilation, m.groups]
+_other_args(m::Conv) = (m.stride, m.pad, m.dilation, m.groups)
+_other_args(m::ConvTranspose) = (m.stride, m.pad, m.outpad, m.dilation, m.groups)
 
 # CPU -> GPU