From f6d8b7da28c1eb2a0e81fbff2439e236312ab005 Mon Sep 17 00:00:00 2001 From: Will Tebbutt Date: Fri, 4 Oct 2024 09:49:42 +0100 Subject: [PATCH] Formatting --- ext/MooncakeCUDAExt.jl | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/ext/MooncakeCUDAExt.jl b/ext/MooncakeCUDAExt.jl index cec1e46f..0ab3eb1f 100644 --- a/ext/MooncakeCUDAExt.jl +++ b/ext/MooncakeCUDAExt.jl @@ -56,17 +56,13 @@ end # Basic rules for operating on CuArrays. @is_primitive( - MinimalCtx, - Tuple{Type{<:CuArray}, UndefInitializer, Vararg{Int, N}} where {N}, + MinimalCtx, Tuple{Type{<:CuArray}, UndefInitializer, Vararg{Int, N}} where {N}, ) function rrule!!( - p::CoDual{Type{P}}, - init::CoDual{UndefInitializer}, - dims::CoDual{Int}... + p::CoDual{Type{P}}, init::CoDual{UndefInitializer}, dims::CoDual{Int}... ) where {P<:CuArray{<:Base.IEEEFloat}} _dims = map(primal, dims) - y = CoDual(P(undef, _dims), P(undef, _dims)) - return y, NoPullback(p, init, dims...) + return CoDual(P(undef, _dims), P(undef, _dims)), NoPullback(p, init, dims...) end end