WIP: adapt to sroa jll #521
Conversation
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?s=60&v=4){kind=small}
| passes = [ | ||
| "inline{default-pipeline=canonicalize max-iterations=4}" | ||
| ] |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| passes = [ | |
| "inline{default-pipeline=canonicalize max-iterations=4}" | |
| ] | |
| passes = ["inline{default-pipeline=canonicalize max-iterations=4}"] |
| push!(passes, "sroa-wrappers") | ||
| push!(passes, "libdevice-funcs-raise") | ||
| push!(passes, "canonicalize") |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| push!(passes, "sroa-wrappers") | |
| push!(passes, "libdevice-funcs-raise") | |
| push!(passes, "canonicalize") | |
| push!(passes, "sroa-wrappers") | |
| push!(passes, "libdevice-funcs-raise") | |
| push!(passes, "canonicalize") |
| return join(passes, | ||
| ',', | ||
| ) |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| return join(passes, | |
| ',', | |
| ) | |
| return join(passes, ',') |
| end | ||
| kern = "lower-kernel{run_init=true toolkitPath=$toolkit cuLaunchKernelPtr=$(cuLaunch[]) cuModuleLoadDataPtr=$(cuModule[]) cuModuleGetFunctionPtr=$(cuFunc[])},symbol-dce" | ||
| if DEBUG_KERNEL[] | ||
| curesulthandler = XLA.Libdl.dlsym(Reactant_jll.libReactantExtra_handle, "ReactantHandleCuResult") |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| curesulthandler = XLA.Libdl.dlsym(Reactant_jll.libReactantExtra_handle, "ReactantHandleCuResult") | |
| curesulthandler = XLA.Libdl.dlsym( | |
| Reactant_jll.libReactantExtra_handle, "ReactantHandleCuResult" | |
| ) |
|
|
||
| function make_tracer( | ||
| seen, | ||
| @nospecialize(prev::Union{Base.ExceptionStack, Core.MethodInstance}), |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| @nospecialize(prev::Union{Base.ExceptionStack, Core.MethodInstance}), | |
| @nospecialize(prev::Union{Base.ExceptionStack,Core.MethodInstance}), |
|
|
||
| # Jax on Github CI dislikes X86 macos | ||
| @static if !Sys.isapple() || Sys.ARCH != :x86_64 | ||
| using PythonCall |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| using PythonCall | |
| using PythonCall |
| @test typeof(result) == ConcreteRNumber{Float32} | ||
| @test result ≈ 6 | ||
| end | ||
| end |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| end |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
Reactant.jl/test/integration/python.jl
Lines 10 to 11 in f289522
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
Reactant.jl/test/integration/python.jl
Lines 13 to 15 in f289522
| offset = 0 | ||
| current_type = T | ||
|
|
||
|
|
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| # Update current_type to the field's type for next iteration | ||
| current_type = fieldtype(current_type, field_idx) | ||
| current_type = tcurrent_type | ||
|
|
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| current_type = tcurrent_type | ||
|
|
||
| end | ||
|
|
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| threaddim = CUDA.CuDim3(threads) | ||
|
|
||
| if convert == Val(true) | ||
| args = recudaconvert.(args) |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| args = recudaconvert.(args) | |
| args = recudaconvert.(args) |
| end | ||
|
|
||
| location = MLIR.IR.Location() | ||
| @assert length(restys) == length(aliases) |
There was a problem hiding this comment.
[JuliaFormatter] reported by reviewdog 🐶
| @assert length(restys) == length(aliases) | |
| @assert length(restys) == length(aliases) |
1 participant
Depends on EnzymeAD/Enzyme-JAX#229
which itself depends on XLA adapting to
llvm/llvm-project#122650
llvm/llvm-project#122646
llvm/llvm-project#122615
llvm/llvm-project#122574
llvm/llvm-project#122572