diff --git a/ext/FluxAMDGPUExt/FluxAMDGPUExt.jl b/ext/FluxAMDGPUExt/FluxAMDGPUExt.jl index 1488672682..85ce0365cb 100644 --- a/ext/FluxAMDGPUExt/FluxAMDGPUExt.jl +++ b/ext/FluxAMDGPUExt/FluxAMDGPUExt.jl @@ -1,5 +1,7 @@ module FluxAMDGPUExt +import ChainRulesCore +import ChainRulesCore: NoTangent import Flux import Flux: adapt_storage, fmap import Flux: DenseConvDims, Conv, ConvTranspose, conv, conv_reshape_bias