Merge pull request #557 from ChrisRackauckas-Claude/fix-fb-serial

Update FastBroadcast extension to Serial dispatch

Author: ChrisRackauckas

Project.toml

@@ -50,7 +50,7 @@ Aqua = "0.8"
 ArrayInterface = "7.16"
 CUDA = "5"
 DocStringExtensions = "0.9.3"
-FastBroadcast = "1.1"
+FastBroadcast = "1.3"
 ForwardDiff = "0.10.38, 1"
 GPUArraysCore = "0.2"
 KernelAbstractions = "0.9.36"

ext/RecursiveArrayToolsFastBroadcastExt.jl

@@ -2,15 +2,15 @@ module RecursiveArrayToolsFastBroadcastExt
 
 using RecursiveArrayTools
 using FastBroadcast
+using FastBroadcast: Serial
 using StaticArraysCore
 
 const AbstractVectorOfSArray = AbstractVectorOfArray{
     T, N, <:AbstractVector{<:StaticArraysCore.SArray},
 } where {T, N}
 
-# Override the base 2-arg path (called by Serial dispatch and direct macro path)
 @inline function FastBroadcast.fast_materialize!(
-        dst::AbstractVectorOfSArray,
+        ::Serial, dst::AbstractVectorOfSArray,
         bc::Broadcast.Broadcasted{S}
     ) where {S}
     if FastBroadcast.use_fast_broadcast(S)