Flux and Octavian

qwerty · November 4, 2022, 9:50am

Does Flux use Octavian when run on cpu?

baggepinnen · November 4, 2022, 10:06am

No, but SimpleChains.jl uses LoopVectorization

PumasAI/SimpleChains.jl/blob/main/src/dense.jl#L182


      
          @inline function dense!(
            f::F,
            C::AbstractVecOrMat{<:Base.HWReal},
            A::AbstractMatrix{<:Base.HWReal},
            B::AbstractVecOrMat{<:Base.HWReal},
            ::True,
            ::True,
          ) where {F}
            Kp1 = ArrayInterface.size(A, StaticInt(2))
            K = Kp1 - StaticInt(1)
            @turbo for n ∈ indices((B, C), 2), m ∈ indices((A, C), 1)
              Cmn = zero(eltype(C))
              for k ∈ 1:K
                Cmn += A[m, k] * B[k, n]
              end
              C[m, n] = f(Cmn + A[m, Kp1])
            end
          end
          @inline function dense!(
            f::F,
            C::AbstractVecOrMat{<:Base.HWReal},

Topic		Replies	Views
Translating tensorflow to Flux and SimpleChains and not getting the same results Machine Learning tensorflow , flux , simplechains	6	547	March 24, 2023
SimpleChains.jl extension to Graph Neural Networks Machine Learning flux , graphneuralnetworks	2	745	May 4, 2022
2 problems with Flux General Usage flux	4	665	January 28, 2020
Errors when trying to compute hessian of flux neural net and jacobian of jacobian with zygote Machine Learning flux , zygote	1	368	July 4, 2022
Flux.jl MethodError ... (::Dense{typeof(identity),... on simple Chain model Machine Learning	3	1136	October 17, 2019

Flux and Octavian

Related topics