From 02a55aca0290bf1612e4896f8983fe304ea1a73b Mon Sep 17 00:00:00 2001
From: Ziyi Yin <ziyi.yin@gatech.edu>
Date: Mon, 27 Mar 2023 22:50:00 -0400
Subject: [PATCH] fix https://github.com/JuliaMath/AbstractFFTs.jl/issues/95

---
 ext/AbstractFFTsChainRulesCoreExt.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ext/AbstractFFTsChainRulesCoreExt.jl b/ext/AbstractFFTsChainRulesCoreExt.jl
index f0c788e6..8b3b69ee 100644
--- a/ext/AbstractFFTsChainRulesCoreExt.jl
+++ b/ext/AbstractFFTsChainRulesCoreExt.jl
@@ -37,7 +37,7 @@ function ChainRulesCore.rrule(::typeof(rfft), x::AbstractArray{<:Real}, dims)
 
     project_x = ChainRulesCore.ProjectTo(x)
     function rfft_pullback(ȳ)
-        x̄ = project_x(brfft(ChainRulesCore.unthunk(ȳ) ./ scale, d, dims))
+	x̄ = project_x(brfft(ChainRulesCore.unthunk(ȳ) ./ typeof(x)(scale), d, dims))
         return ChainRulesCore.NoTangent(), x̄, ChainRulesCore.NoTangent()
     end
     return y, rfft_pullback