Merge pull request #626 from Jutho/patch-1

restrict strided array multiplication `rrule`
JuliaDiff · May 30, 2022 · 3d381ab · 3d381ab
2 parents d0bcfc5 + 542abbf
commit 3d381ab
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/rulesets/Base/arraymath.jl b/src/rulesets/Base/arraymath.jl
@@ -46,9 +46,9 @@ end
 # https://github.com/JuliaDiff/ChainRulesCore.jl/issues/411
 function rrule(
     ::typeof(*),
-    A::StridedMatrix{<:CommutativeMulNumber},
-    B::StridedVecOrMat{<:CommutativeMulNumber},
-)
+    A::StridedMatrix{T},
+    B::StridedVecOrMat{T},
+) where {T<:CommutativeMulNumber}
     function times_pullback(ȳ)
         Ȳ = unthunk(ȳ)
         dA = InplaceableThunk(