chore: formatting

LuxDL · Jul 13, 2024 · c15f291 · c15f291
1 parent 380b841
commit c15f291
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ext/LuxLibcuDNNExt/LuxLibcuDNNExt.jl b/ext/LuxLibcuDNNExt/LuxLibcuDNNExt.jl
@@ -35,6 +35,7 @@ end
 
 function CRC.rrule(::typeof(LuxLib.batchnorm_cudnn), running_mean, running_var, scale,
         bias, x, momentum, epsilon, t::Val{training}) where {training}
+    # TODO: Transition this to an error in the future
     !training && @warn "`training=Val(false)` but gradient was called." maxlog=1
     y, xmean, xivar = LuxLib.batchnorm_cudnn(
         running_mean, running_var, scale, bias, x, momentum, epsilon, t)

diff --git a/src/impl/fused_conv.jl b/src/impl/fused_conv.jl
@@ -66,8 +66,8 @@ function __∇conv_filter(
     return ∇conv_filter(x, y, cdims)
 end
 
-function __conv_bias_act(x_::AbstractArray{xT}, weight_::AbstractArray{wT},
-        cdims::ConvDims, bias_::Optional{<:AbstractArray}, act::F) where {xT, wT, F}
+function __conv_bias_act(x_::AbstractArray{xT}, weight_::AbstractArray{wT}, cdims::ConvDims,
+        bias_::Optional{<:AbstractArray}, act::F) where {xT, wT, F}
     dev = get_device_type((x_, weight_, bias_))
     x, weight = __get_conv_input_weight(dev, xT, wT, x_, weight_)
     bias = _ofeltype_array(eltype(x), bias_)