fix: explicit imports

LuxDL · Jul 13, 2024 · ea75484 · ea75484
1 parent 6ea9a63
commit ea75484
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ext/LuxLibForwardDiffExt.jl b/ext/LuxLibForwardDiffExt.jl
@@ -2,7 +2,7 @@ module LuxLibForwardDiffExt
 
 using ForwardDiff: ForwardDiff
 using LuxLib: LuxLib
-using LuxDeviceUtils: AbstractLuxDevice, AbstractLuxGPUDevice
+using LuxDeviceUtils: AbstractLuxGPUDevice
 using NNlib: NNlib
 
 LuxLib.__has_dual(::ForwardDiff.Dual) = true
@@ -80,6 +80,6 @@ end
 
 LuxLib.__value(x::ForwardDiff.Dual) = ForwardDiff.value(x)
 LuxLib.__value(x::AbstractArray{<:ForwardDiff.Dual}) = ForwardDiff.value.(x)
-LuxLib.__value(::Type{<:ForwardDiff.Dual{T}}) where {T} = LuxLib.__value(T)
+LuxLib.__value(::Type{<:ForwardDiff.Dual{Tag, T}}) where {Tag, T} = LuxLib.__value(T)
 
 end
diff --git a/src/api/conv.jl b/src/api/conv.jl
@@ -38,7 +38,8 @@ for (check, fop) in (
     (false, :_fused_conv_bias_activation_impl), (true, :_generic_conv_bias_activation))
     @eval function fused_conv_bias_activation(
             σ::F, ::Val{$(check)}, weight::AbstractArray{<:Number, N},
-            x::AbstractArray{<:Number, N}, b::Nothing, cdims::ConvDims) where {F, N}
+            x::AbstractArray{<:Number, N},
+            b::Optional{<:AbstractArray}, cdims::ConvDims) where {F, N}
         return $(fop)(σ, weight, x, b, cdims)
     end
 end