Failed to precompile `SymbolicsLuxExt`

[SebastianM-C]

I'm seeing

Failed to precompile SymbolicsLuxExt [e6b0080e-84b4-5885-ba4d-e6e9d1269bd5] to "/home/sebastian/.julia/compiled/v1.11/SymbolicsLuxExt/jl_ebxtxg".
ERROR: LoadError: MethodError: no method matching __iterspace(::Type{AbstractArray})
The function `__iterspace` exists, but no method is defined for this combination of argument types.

Closest candidates are:
  __iterspace(::KernelAbstractions.CompilerMetadata)
   @ KernelAbstractions ~/.julia/packages/KernelAbstractions/PjD0d/src/compiler.jl:23

Stacktrace:
  [1] ndims(ctx::Type)
    @ KernelAbstractions ~/.julia/packages/KernelAbstractions/PjD0d/src/compiler.jl:33
  [2] (::Symbolics.var"#2#11"{UnionAll})(elT::Expr)
    @ Symbolics ~/.julia/packages/Symbolics/EftHA/src/wrapper-types.jl:120
  [3] (::Symbolics.var"#type_options#10"{Module, Bool})(arg::Expr)
    @ Symbolics ~/.julia/packages/Symbolics/EftHA/src/wrapper-types.jl:125
  [4] iterate
    @ ./generator.jl:48 [inlined]
  [5] collect_to!(dest::Vector{Tuple{DataType, Expr}}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, offs::Int64, st::Int64)
    @ Base ./array.jl:849
  [6] collect_to_with_first!(dest::Vector{Tuple{DataType, Expr}}, v1::Tuple{DataType, Expr}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, st::Int64)
    @ Base ./array.jl:827
  [7] _collect(c::Vector{Any}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, ::Base.EltypeUnknown, isz::Base.HasShape{1})
    @ Base ./array.jl:821
  [8] collect_similar(cont::Vector{Any}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}})
    @ Base ./array.jl:720
  [9] map(f::Function, A::Vector{Any})
    @ Base ./abstractarray.jl:3371
 [10] wrap_func_expr(mod::Module, expr::Expr, wrap_arrays::Bool)
    @ Symbolics ~/.julia/packages/Symbolics/EftHA/src/wrapper-types.jl:140
 [11] var"@wrapped"(__source__::LineNumberNode, __module__::Module, expr::Any, wrap_arrays::Any)
    @ Symbolics ~/.julia/packages/Symbolics/EftHA/src/wrapper-types.jl:175
 [12] include
    @ ./Base.jl:557 [inlined]
 [13] include_package_for_output(pkg::Base.PkgId, input::String, depot_path::Vector{String}, dl_load_path::Vector{String}, load_path::Vector{String}, concrete_deps::Vector{Pair{Base.PkgId, UInt128}}, source::Nothing)
    @ Base ./loading.jl:2881
 [14] top-level scope
    @ stdin:6
in expression starting at /home/sebastian/.julia/packages/Symbolics/EftHA/src/register.jl:109
in expression starting at /home/sebastian/.julia/packages/Symbolics/EftHA/ext/SymbolicsLuxExt.jl:1

with

  [b2108857] Lux v1.5.2
  [0c5d862f] Symbolics v6.24.0

I don't think that the tests for the extensions are ran in CI. Edit: #1414 should fix that

[SebastianM-C]

See https://github.com/JuliaSymbolics/Symbolics.jl/actions/runs/12897778565/job/35963566607?pr=1414 for the tests now running (erroring) in ci.

[B-LIE]

I get the same error, with details as below. I've had this error before. It seems to depend on which packages one has installed. Is it NonlinearSolve that causes the problem? Or Lux? Or something else? [Based on previous discussion on the topic...]

julia> err
PkgPrecompileError: The following 1 direct dependency failed to precompile:

SymbolicsLuxExt

Failed to precompile SymbolicsLuxExt [e6b0080e-84b4-5885-ba4d-e6e9d1269bd5] to "C:\\Users\\Bernt\\.julia\\compiled\\v1.11\\SymbolicsLuxExt\\jl_A30B.tmp".
ERROR: LoadError: MethodError: no method matching __iterspace(::Type{AbstractArray})
The function `__iterspace` exists, but no method is defined for this combination of argument types.

Closest candidates are:
  __iterspace(::KernelAbstractions.CompilerMetadata)
   @ KernelAbstractions C:\Users\Bernt\.julia\packages\KernelAbstractions\PjD0d\src\compiler.jl:23

Stacktrace:
  [1] ndims(ctx::Type)
    @ KernelAbstractions C:\Users\Bernt\.julia\packages\KernelAbstractions\PjD0d\src\compiler.jl:33
  [2] (::Symbolics.var"#2#11"{UnionAll})(elT::Expr)
    @ Symbolics C:\Users\Bernt\.julia\packages\Symbolics\EftHA\src\wrapper-types.jl:120
  [3] (::Symbolics.var"#type_options#10"{Module, Bool})(arg::Expr)
    @ Symbolics C:\Users\Bernt\.julia\packages\Symbolics\EftHA\src\wrapper-types.jl:125
  [4] iterate
    @ .\generator.jl:48 [inlined]
  [5] collect_to!(dest::Vector{Tuple{DataType, Expr}}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, offs::Int64, st::Int64)
    @ Base .\array.jl:849
  [6] collect_to_with_first!(dest::Vector{Tuple{DataType, Expr}}, v1::Tuple{DataType, Expr}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, st::Int64)
    @ Base .\array.jl:827
  [7] _collect(c::Vector{Any}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}}, ::Base.EltypeUnknown, isz::Base.HasShape{1})
    @ Base .\array.jl:821
  [8] collect_similar(cont::Vector{Any}, itr::Base.Generator{Vector{Any}, Symbolics.var"#type_options#10"{Module, Bool}})
    @ Base .\array.jl:720
  [9] map(f::Function, A::Vector{Any})
    @ Base .\abstractarray.jl:3371
 [10] wrap_func_expr(mod::Module, expr::Expr, wrap_arrays::Bool)
    @ Symbolics C:\Users\Bernt\.julia\packages\Symbolics\EftHA\src\wrapper-types.jl:140
 [11] var"@wrapped"(__source__::LineNumberNode, __module__::Module, expr::Any, wrap_arrays::Any)
    @ Symbolics C:\Users\Bernt\.julia\packages\Symbolics\EftHA\src\wrapper-types.jl:175
 [12] include
    @ .\Base.jl:557 [inlined]
 [13] include_package_for_output(pkg::Base.PkgId, input::String, depot_path::Vector{String}, dl_load_path::Vector{String}, load_path::Vector{String}, concrete_deps::Vector{Pair{Base.PkgId, UInt128}}, source::Nothing)
    @ Base .\loading.jl:2881
 [14] top-level scope
    @ stdin:6
in expression starting at C:\Users\Bernt\.julia\packages\Symbolics\EftHA\src\register.jl:109
in expression starting at C:\Users\Bernt\.julia\packages\Symbolics\EftHA\ext\SymbolicsLuxExt.jl:1
in expression starting at stdin:

[ChrisRackauckas]

@avik-pal why is kernelabstractions getting into this?

[avik-pal]

Seems like a type piracy in KA? KA is a dep in Lux

[ChrisRackauckas]

Yeah I found that out later today... https://github.com/JuliaGPU/KernelAbstractions.jl/pull/551/files#r1925068505

[trafficfan]

Meet the same issue ，hoping for a fix

[ChrisRackauckas]

We're going to put an upper bound on KernalAbstractions until @vchuravy can get around to a fix, since it's a KernelAbstractions.jl piracy thing and not something in our libraries it can happen when any package adds KA, so the only real fix is to bound.

[ChrisRackauckas]

Actually looks like he already got to it: JuliaGPU/KernelAbstractions.jl#557 release is probably incoming.

[ChrisRackauckas]

v0.9.33 should do it: https://github.com/JuliaGPU/KernelAbstractions.jl/releases/tag/v0.9.33 can someone confirm that if you update to the versions released 40 minutes ago (i.e. just do a ]up) that the problem is fixed? If so then this is closed.

[B-LIE]

Fixed on my computer.

[ChrisRackauckas]

Thanks @vchuravy for getting to this!