diff --git a/ext/LuxLibForwardDiffExt.jl b/ext/LuxLibForwardDiffExt.jl index 24622cdc..f3c54eac 100644 --- 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