make BatchedAdjOrTrans return correct BroadcastStyle (#424)

chengchingwen · web-flow · commit c9faa6407c99 · 2022-06-26T12:04:51.000+08:00
diff --git a/src/batched/batchedadjtrans.jl b/src/batched/batchedadjtrans.jl
@@ -102,3 +102,5 @@ end
 
 adapt_structure(to, x::BatchedAdjoint) = BatchedAdjoint(adapt(to, parent(x)))
 adapt_structure(to, x::BatchedTranspose) = BatchedTranspose(adapt(to, parent(x)))
+
+Broadcast.BroadcastStyle(::Type{<:BatchedAdjOrTrans{T, S}}) where {T, S} = Broadcast.BroadcastStyle(S)
