Interp2 #365

wsmoses · 2024-12-11T18:29:45Z

No description provided.

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

github-actions · 2024-12-14T07:05:00Z

src/TracedRArray.jl

+import ..ancestor
+import ReactantCore
+import ..TracedUtils: materialize_traced_array
+import GPUArraysCore


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

import GPUArraysCore

using GPUArraysCore: GPUArraysCore

github-actions · 2024-12-14T07:05:00Z

src/TracedRArray.jl

-get_ancestor_indices(::TracedRArray, indices...) = indices
-function get_ancestor_indices(x::WrappedTracedRArray, indices...)
-    return get_ancestor_indices(parent(x), Base.reindex(parentindices(x), indices)...)
-end



[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T07:05:00Z

src/TracedRArray.jl


 function Base.getindex(
    a::TracedRArray{T,N}, index::Vararg{Union{Int,TracedRNumber{Int}},N}
 ) where {T,N}
    GPUArraysCore.assertscalar("getindex(::TracedRArray, ::Vararg{Int, N})")

-    start_indices = [promote_to(TracedRNumber{Int}, i - 1).mlir_data for i in index]
+    start_indices = [TracedUtils.promote_to(TracedRNumber{Int}, i - 1).mlir_data for i in index]


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

start_indices = [TracedUtils.promote_to(TracedRNumber{Int}, i - 1).mlir_data for i in index]

start_indices = [

TracedUtils.promote_to(TracedRNumber{Int}, i - 1).mlir_data for i in index

]

github-actions · 2024-12-14T07:05:00Z

src/TracedRArray.jl

+        (TracedUtils.promote_to(TracedRNumber{Int}, i isa Colon ? 1 : first(i)) - 1).mlir_data for
        i in indices


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

(TracedUtils.promote_to(TracedRNumber{Int}, i isa Colon ? 1 : first(i)) - 1).mlir_data for

i in indices

(

TracedUtils.promote_to(TracedRNumber{Int}, i isa Colon ? 1 : first(i)) - 1

).mlir_data for i in indices

github-actions · 2024-12-14T07:05:00Z

src/TracedRArray.jl

-    return traced2_result
-end
+TracedUtils.promote_to(::Type{TracedRArray{T,N}}, rhs) where {T,N} = TracedRArray{T,N}(rhs)
+TracedUtils.promote_to(::TracedRArray{T,N}, rhs) where {T,N} = TracedUtils.promote_to(TracedRArray{T,N}, rhs)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

TracedUtils.promote_to(::TracedRArray{T,N}, rhs) where {T,N} = TracedUtils.promote_to(TracedRArray{T,N}, rhs)

function TracedUtils.promote_to(::TracedRArray{T,N}, rhs) where {T,N}

return TracedUtils.promote_to(TracedRArray{T,N}, rhs)

end

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

JuliaFormatter

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 490 in f4349a9

github-actions · 2024-12-14T08:31:26Z

src/TracedRNumber.jl

 end

-Base.float(x::TracedRNumber{T}) where {T} = promote_to(TracedRNumber{float(T)}, x)
+Base.float(x::TracedRNumber{T}) where {T} = TracedUtils.promote_to(TracedRNumber{float(T)}, x)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

Base.float(x::TracedRNumber{T}) where {T} = TracedUtils.promote_to(TracedRNumber{float(T)}, x)

function Base.float(x::TracedRNumber{T}) where {T}

return TracedUtils.promote_to(TracedRNumber{float(T)}, x)

end

github-actions · 2024-12-14T08:31:26Z

src/TracedUtils.jl

+
+struct TypeCast{T<:ReactantPrimitive} <: Function end
+
+(::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2} = TracedUtils.promote_to(TracedRNumber{T}, x)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

(::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2} = TracedUtils.promote_to(TracedRNumber{T}, x)

function (::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2}

return TracedUtils.promote_to(TracedRNumber{T}, x)

end

github-actions · 2024-12-14T08:31:26Z

src/TracedUtils.jl

+    end
+
+    seen_results = OrderedIdDict()
+    traced2_result = Reactant.make_tracer(seen_results, result, (), Reactant.TracedSetPath; tobatch=OutShape)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

traced2_result = Reactant.make_tracer(seen_results, result, (), Reactant.TracedSetPath; tobatch=OutShape)

traced2_result = Reactant.make_tracer(

seen_results, result, (), Reactant.TracedSetPath; tobatch=OutShape

)

github-actions · 2024-12-14T08:52:51Z

lib/ReactantCore/src/ReactantCore.jl

@@ -153,15 +153,15 @@ function trace_for(mod, expr)

    all_syms = Expr(:tuple, counter, external_syms...)
    args_init = Expr(
-        :tuple, :(Reactant.promote_to(Reactant.TracedRNumber{Int}, 0)), external_syms...
+        :tuple, :(Reactant.TracedUtils.promote_to(Reactant.TracedRNumber{Int}, 0)), external_syms...


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

:tuple, :(Reactant.TracedUtils.promote_to(Reactant.TracedRNumber{Int}, 0)), external_syms...

:tuple,

:(Reactant.TracedUtils.promote_to(Reactant.TracedRNumber{Int}, 0)),

external_syms...,

github-actions · 2024-12-14T08:52:51Z

src/linear_algebra.jl

+using ..TracedUtils:
+    get_mlir_data,
+    materialize_traced_array,
+    set_mlir_data!


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

using ..TracedUtils:

get_mlir_data,

materialize_traced_array,

set_mlir_data!

using ..TracedUtils: get_mlir_data, materialize_traced_array, set_mlir_data!

github-actions · 2024-12-14T09:28:45Z

ext/ReactantNNlibExt.jl

@@ -329,7 +332,7 @@ function NNlib.gather!(dst::TracedRArray, src::AnyTracedRArray, idxs::AbstractAr
    start_sizes = ntuple(i -> size(src, i), dims)
    results = map(CartesianIndices(idxs)) do k
        res = @allowscalar src[colons..., Tuple(idxs[k])...]
-        res isa TracedRNumber && (res = Reactant.broadcast_to_size(res, (1,)))
+        res isa TracedRNumber && (res = Reactant.TracedUtils.broadcast_to_size(res, (1,)))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

res isa TracedRNumber && (res = Reactant.TracedUtils.broadcast_to_size(res, (1,)))

res isa TracedRNumber &&

(res = Reactant.TracedUtils.broadcast_to_size(res, (1,)))

github-actions · 2024-12-14T09:28:45Z

src/Tracing.jl

@@ -498,14 +498,14 @@ function make_tracer(
            return ConcreteRNumber(prev)
        else
            if mode == TracedTrack
-                res = TracedRNumber{RT}((path,), broadcast_to_size(prev, ()).mlir_data)
+                res = TracedRNumber{RT}((path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

res = TracedRNumber{RT}((path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data)

res = TracedRNumber{RT}(

(path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data

)

github-actions · 2024-12-14T09:28:45Z

src/Tracing.jl

                if !haskey(seen, prev)
                    return seen[prev] = res
                end
                return res
            elseif mode == TracedSetPath
                haskey(seen, prev) && return seen[prev]
-                res = TracedRNumber{RT}((path,), broadcast_to_size(prev, ()).mlir_data)
+                res = TracedRNumber{RT}((path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

res = TracedRNumber{RT}((path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data)

res = TracedRNumber{RT}(

(path,), TracedUtils.broadcast_to_size(prev, ()).mlir_data

)

github-actions · 2024-12-14T09:28:45Z

src/linear_algebra.jl

@@ -61,7 +80,7 @@ end
 function LinearAlgebra.triu!(@nospecialize(X::TracedRArray{T,2}), k::Integer) where {T}
    iota_1 = Ops.iota(Int64, [size(X)...]; iota_dimension=1)
    iota_2 = Ops.subtract(
-        Ops.iota(Int64, [size(X)...]; iota_dimension=2), broadcast_to_size(k, size(X))
+        Ops.iota(Int64, [size(X)...]; iota_dimension=2), TracedUtils.broadcast_to_size(k, size(X))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

Ops.iota(Int64, [size(X)...]; iota_dimension=2), TracedUtils.broadcast_to_size(k, size(X))

Ops.iota(Int64, [size(X)...]; iota_dimension=2),

TracedUtils.broadcast_to_size(k, size(X)),

github-actions · 2024-12-14T09:28:45Z

src/linear_algebra.jl

@@ -71,7 +90,7 @@ end
 function LinearAlgebra.tril!(@nospecialize(X::TracedRArray{T,2}), k::Integer) where {T}
    iota_1 = Ops.iota(Int64, [size(X)...]; iota_dimension=1)
    iota_2 = Ops.subtract(
-        Ops.iota(Int64, [size(X)...]; iota_dimension=2), broadcast_to_size(k, size(X))
+        Ops.iota(Int64, [size(X)...]; iota_dimension=2), TracedUtils.broadcast_to_size(k, size(X))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

Ops.iota(Int64, [size(X)...]; iota_dimension=2), TracedUtils.broadcast_to_size(k, size(X))

Ops.iota(Int64, [size(X)...]; iota_dimension=2),

TracedUtils.broadcast_to_size(k, size(X)),

github-actions · 2024-12-14T10:01:20Z

src/linear_algebra.jl

@@ -115,7 +134,7 @@ function LinearAlgebra.diag(x::AnyTracedRArray{T,2}, k::Integer=0) where {T}
    )
    #! format: on

-    slice_sizes = get_mlir_data(Reactant.promote_to(TracedRArray{Int,1}, [1, 1]))
+    slice_sizes = get_mlir_data(Reactant.TracedUtils.promote_to(TracedRArray{Int,1}, [1, 1]))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

slice_sizes = get_mlir_data(Reactant.TracedUtils.promote_to(TracedRArray{Int,1}, [1, 1]))

slice_sizes = get_mlir_data(

Reactant.TracedUtils.promote_to(TracedRArray{Int,1}, [1, 1])

)

github-actions · 2024-12-14T10:01:20Z

src/linear_algebra.jl

@@ -139,6 +158,8 @@ function LinearAlgebra.diagm(m::Integer, n::Integer, v::AnyTracedRArray{T,1}) wh

    mat = (v .+ zero(v)') .* diag_indicator
    return Ops.pad(
-        mat, promote_to(TracedRNumber{T}, 0); high=[m - length(v), n - length(v)]
+        mat, TracedUtils.promote_to(TracedRNumber{T}, 0); high=[m - length(v), n - length(v)]


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

mat, TracedUtils.promote_to(TracedRNumber{T}, 0); high=[m - length(v), n - length(v)]

mat,

TracedUtils.promote_to(TracedRNumber{T}, 0);

high=[m - length(v), n - length(v)],

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

JuliaFormatter

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Lines 510 to 513 in 70c3951

    
           push!( 
        
               overdubbed_code, 
        
               Core.ReturnNode(Core.SSAValue(length(overdubbed_code))) 
        
           )

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 526 in 70c3951

github-actions

Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit

JuliaFormatter

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 436 in f839a0b

iter_args = min(n_actual_args, n_method_args-1)

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 438 in f839a0b

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 448 in f839a0b

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 450 in f839a0b

    
           push!(overdubbed_code, Expr(:call, safe_print, "fn arg["*string(length(fn_args))*"]", fn_args[end]))

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 455 in f839a0b

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Lines 470 to 472 in f839a0b

    
           push!( 
        
               overdubbed_code, trailing_arguments 
        
           )

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 476 in f839a0b

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 478 in f839a0b

    
           push!(overdubbed_code, Expr(:call, safe_print, "fn arg["*string(length(fn_args))*"]", fn_args[end]))

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 484 in f839a0b

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 500 in f839a0b

    
           res = Core._call_in_world_total(world, make_oc, octup, rt, src, ocnargs, ocva, args[1].instance)::Core.OpaqueClosure

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Lines 504 to 514 in f839a0b

    
           push!(overdubbed_code, 
        
               Expr(:call, 
        
                   make_oc, 
        
                   octup, 
        
                   rt, 
        
                   src, 
        
                   ocnargs, 
        
                   ocva, 
        
                   farg 
        
                   ) 
        
                   )

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Lines 519 to 526 in f839a0b

    
           push!( 
        
               overdubbed_code, 
        
               Expr( 
        
                   :(call), 
        
                   oc, 
        
                   fn_args[2:end]... 
        
               ), 
        
           )

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Lines 530 to 533 in f839a0b

    
           push!( 
        
               overdubbed_code, 
        
               Core.ReturnNode(Core.SSAValue(length(overdubbed_code))) 
        
           )

[JuliaFormatter] _{reported by reviewdog 🐶}

Reactant.jl/src/utils.jl

Line 546 in f839a0b

github-actions · 2024-12-14T16:29:46Z

src/TracedRArray.jl

+    rdims = Int64[]
+
+    if dims == (:)
+        for i in 0:(N-1)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

for i in 0:(N-1)

for i in 0:(N - 1)

github-actions · 2024-12-14T16:29:47Z

src/TracedRArray.jl

    else
-        Int64[i - 1 for i in dims]
+        for i in dims
+            push!(rdims, i-1)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(rdims, i-1)

push!(rdims, i - 1)

github-actions · 2024-12-14T16:29:47Z

src/TracedRArray.jl

+    toonedims = Int[]
+    outdims = Int[]
+    for i in 1:N
+        tmp = if in(i-1, rdims)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

tmp = if in(i-1, rdims)

tmp = if in(i - 1, rdims)

github-actions · 2024-12-14T16:29:47Z

src/TracedRArray.jl

@@ -471,19 +283,19 @@ function Base.mapreducedim!(
    @nospecialize(R::TracedRArray),
    A::Base.AbstractArrayOrBroadcasted,
 )
-    tmp = broadcast_to_size(Base.mapreduce(f, op, A; dims=1), (1, size(R)[2:end]...))
+    tmp = TracedUtils.broadcast_to_size(Base.mapreduce(f, op, A; dims=1), (1, size(R)[2:end]...))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

tmp = TracedUtils.broadcast_to_size(Base.mapreduce(f, op, A; dims=1), (1, size(R)[2:end]...))

tmp = TracedUtils.broadcast_to_size(

Base.mapreduce(f, op, A; dims=1), (1, size(R)[2:end]...)

)

github-actions · 2024-12-14T16:29:47Z

src/TracedRArray.jl

    A.mlir_data = bcast.mlir_data
    return A
 end

 function Base.fill!(A::TracedRArray{T,N}, x::TracedRNumber{T2}) where {T,N,T2}
-    bcast = broadcast_to_size(promote_to(TracedRNumber{T}, x), size(A))
+    bcast = TracedUtils.broadcast_to_size(TracedUtils.promote_to(TracedRNumber{T}, x), size(A))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

bcast = TracedUtils.broadcast_to_size(TracedUtils.promote_to(TracedRNumber{T}, x), size(A))

bcast = TracedUtils.broadcast_to_size(

TracedUtils.promote_to(TracedRNumber{T}, x), size(A)

)

github-actions · 2024-12-14T16:29:48Z

src/utils.jl

+        mi = ccall(:jl_specializations_get_linfo, Ref{Core.MethodInstance},
+                   (Any, Any, Any), match.method, match.spec_types, match.sparams)
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

mi = ccall(:jl_specializations_get_linfo, Ref{Core.MethodInstance},

(Any, Any, Any), match.method, match.spec_types, match.sparams)

mi = ccall(

:jl_specializations_get_linfo,

Ref{Core.MethodInstance},

(Any, Any, Any),

match.method,

match.spec_types,

match.sparams,

)

github-actions · 2024-12-14T16:29:48Z

src/utils.jl

+        src = copy(ci)
+        src.slotnames = Any[:call_with_reactant, REDUB_ARGUMENTS_NAME]
+
+        src.edges = Any[ccall(:jl_method_table_for, Any, (Any,), sig)::Core.MethodTable, sig]


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

src.edges = Any[ccall(:jl_method_table_for, Any, (Any,), sig)::Core.MethodTable, sig]

src.edges = Any[

ccall(:jl_method_table_for, Any, (Any,), sig)::Core.MethodTable, sig

]

github-actions · 2024-12-14T16:29:48Z

src/utils.jl

+
+        expr_fn = Core.SSAValue(length(overdubbed_code))
+
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T16:29:48Z

src/utils.jl

+
+        push!(overdubbed_code, :(2:$expr_lastindex))


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(overdubbed_code, :(2:$expr_lastindex))

push!(overdubbed_code, :(2:($expr_lastindex)))

github-actions · 2024-12-14T16:29:49Z

src/utils.jl

+        push!(
+            overdubbed_code,
+            Core.ReturnNode(Core.SSAValue(length(overdubbed_code)))
        )


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(

overdubbed_code,

Core.ReturnNode(Core.SSAValue(length(overdubbed_code)))

)

push!(overdubbed_code, Core.ReturnNode(Core.SSAValue(length(overdubbed_code))))

github-actions · 2024-12-14T19:29:47Z

src/utils.jl

+            Core.Compiler.ipo_dataflow_analysis!(interp, ir, caller)
+        end
+    end
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T19:29:47Z

src/utils.jl

        end
    end
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T19:29:47Z

src/utils.jl

-            body=MLIR.IR.Region(),
-            sym_visibility,
+    src = Core.Compiler.ir_to_codeinf!(opt)
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T19:29:47Z

src/utils.jl

+    n_actual_args = length(redub_arguments)
+
+    tys = []
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

github-actions · 2024-12-14T19:29:47Z

src/utils.jl

+
+    iter_args = n_actual_args
+    if method.isva
+        iter_args = min(n_actual_args, n_method_args-1)


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

iter_args = min(n_actual_args, n_method_args-1)

iter_args = min(n_actual_args, n_method_args - 1)

github-actions · 2024-12-14T19:29:48Z

src/utils.jl

+    # Opaque closures also require takign the function argument. We can work around the latter
+    # if the function is stateless. But regardless, to work around this we sadly create/compile the opaque closure
+    oc = if false && Base.issingletontype(args[1])
+        res = Core._call_in_world_total(world, make_oc, octup, rt, src, ocnargs, ocva, args[1].instance)::Core.OpaqueClosure


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

res = Core._call_in_world_total(world, make_oc, octup, rt, src, ocnargs, ocva, args[1].instance)::Core.OpaqueClosure

res = Core._call_in_world_total(

world, make_oc, octup, rt, src, ocnargs, ocva, args[1].instance

)::Core.OpaqueClosure

github-actions · 2024-12-14T19:29:48Z

src/utils.jl

+        push!(overdubbed_code,
+            Expr(:call,
+                make_oc,
+                octup,
+                rt,
+                src,
+                ocnargs,
+                ocva,
+                farg
+                )
+                )


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(overdubbed_code,

Expr(:call,

make_oc,

octup,

rt,

src,

ocnargs,

ocva,

farg

)

)

push!(overdubbed_code, Expr(:call, make_oc, octup, rt, src, ocnargs, ocva, farg))

github-actions · 2024-12-14T19:29:48Z

src/utils.jl

+    push!(
+        overdubbed_code,
+        Expr(
+            :(call),
+            oc,
+            fn_args[2:end]...
+        ),
+    )


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(

overdubbed_code,

Expr(

:(call),

oc,

fn_args[2:end]...

),

)

push!(overdubbed_code, Expr(:(call), oc, fn_args[2:end]...))

github-actions · 2024-12-14T19:29:48Z

src/utils.jl

+    push!(
+        overdubbed_code,
+        Core.ReturnNode(Core.SSAValue(length(overdubbed_code)))
+    )


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

push!(

overdubbed_code,

Core.ReturnNode(Core.SSAValue(length(overdubbed_code)))

)

push!(overdubbed_code, Core.ReturnNode(Core.SSAValue(length(overdubbed_code))))

github-actions · 2024-12-14T19:29:48Z

src/utils.jl

+    code_info.codelocs = overdubbed_codelocs
+    code_info.ssavaluetypes = length(overdubbed_code)
+    code_info.ssaflags = [0x00 for _ in 1:length(overdubbed_code)] # XXX we need to copy flags that are set for the original code
+


[JuliaFormatter] _{reported by reviewdog 🐶}

Suggested change

commit 65e9976 Author: William Moses <[email protected]> Date: Sat Dec 14 14:05:03 2024 -0600 Interp2 (EnzymeAD#365) * WIP: kernels * more files * fix * wip * wqtmp * wip * inc * continuing * wip * more work * inf rec * fix * overload working * continuing * continuing * push * fix `call_with_reactant_generator` for Julia 1.11 (EnzymeAD#359) * conversion * continuing * Cleanup * Apply suggestions from code review Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> * Delete test/cuda.jl * fixup * Apply suggestions from code review Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> * fix apply * indep of change * minor fix in name * Update utils.jl * Interp take 2 * continuing adentures * delcode * fix * tmp * make * fix * cleanup * continuing * more working * further simplify * fx * more improvements * minus show * less prints * even fewer * confusion * tmp * force clean * force oc * clean * Rewrite * fixup * fix * fix * fix * fixup * fix * wip * safe prints * fix * fix * stackoverflow * cleanup * dyindex * rt * continue * clean * fix * fix * fix * fix * fixup * fix * fix * capture oc * compile perf * v1.11 fix * other way 'round * formatting --------- Co-authored-by: William Moses <[email protected]> Co-authored-by: jumerckx <[email protected]> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: jumerckx <[email protected]> commit 73899f5 Author: Avik Pal <[email protected]> Date: Sat Dec 14 14:58:47 2024 +0530 fix: include files if they end with .jl (EnzymeAD#377) commit 9f96c09 Author: Sergio Sánchez Ramírez <[email protected]> Date: Fri Dec 13 23:12:43 2024 +0100 Run CI on aarch64 (EnzymeAD#350) * Run CI on aarch64 * use julia pipeline for aarch64-linux * fix var * exclude aarch64-linux jobs from github ci commit b56e661 Author: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Date: Fri Dec 13 08:58:51 2024 +0530 chore: format code (EnzymeAD#371) Co-authored-by: mofeing <[email protected]> commit 311498b Author: Anton Oresten <[email protected]> Date: Thu Dec 12 05:41:39 2024 +0100 feat: define outer `repeat` method for `TracedRArray` (EnzymeAD#361) * Add repeat method * Add repeat tests * Update test/basic.jl * Update src/TracedRArray.jl commit 8b90501 Author: Avik Pal <[email protected]> Date: Thu Dec 12 10:11:00 2024 +0530 fix: ensure printing of wrapped ConcreteRArrays goes through our show (EnzymeAD#367) * fix: ensure printing of wrapped ConcreteRArrays goes through our show * fix: allow wrapped arrays in mapreduce commit ea97be3 Author: Sergio Sánchez Ramírez <[email protected]> Date: Wed Dec 11 22:02:14 2024 +0100 Ignore versioned manifests

William Moses and others added 30 commits December 10, 2024 22:21

WIP: kernels

7159756

more files

f891bde

fix

14174db

wip

e401e4a

wqtmp

e7bc318

wip

ae6c7c6

inc

b8f206f

continuing

c414601

wip

9138a36

more work

c2ca4cb

inf rec

da328d3

fix

ad4d05b

overload working

750661a

continuing

12dec6c

continuing

a6cd104

push

db6e37b

fix call_with_reactant_generator for Julia 1.11 (#359)

8831f4d

conversion

e2ffe87

continuing

364823a

Cleanup

ff729ce

Apply suggestions from code review

3bd5608

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Delete test/cuda.jl

5e33afb

fixup

17c2f72

Apply suggestions from code review

4807a79

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

fix apply

51286bd

indep of change

bd40b69

minor fix in name

5b3329c

Update utils.jl

4af4a00

Interp take 2

8379f05

continuing adentures

246ec4e

dyindex

b17e75f

github-actions bot reviewed Dec 14, 2024

View reviewed changes

wsmoses added 3 commits December 14, 2024 02:16

rt

2cff76e

continue

1d0cb8e

clean

f4349a9

github-actions bot reviewed Dec 14, 2024

View reviewed changes

wsmoses added 2 commits December 14, 2024 03:28

fix

a4ae31a

fix

0dbe20f

github-actions bot reviewed Dec 14, 2024

View reviewed changes

wsmoses added 2 commits December 14, 2024 03:47

fix

1c45d7e

fix

1ffb366

github-actions bot reviewed Dec 14, 2024

View reviewed changes

wsmoses added 2 commits December 14, 2024 03:52

fixup

3887575

fix

873e46b

github-actions bot reviewed Dec 14, 2024

View reviewed changes

fix

21244db

github-actions bot reviewed Dec 14, 2024

View reviewed changes

capture oc

70c3951

github-actions bot reviewed Dec 14, 2024

View reviewed changes

compile perf

f839a0b

github-actions bot reviewed Dec 14, 2024

View reviewed changes

wsmoses mentioned this pull request Dec 14, 2024

Interpreter fixes (prerequisite to CUDA support) #364

Closed

v1.11 fix

8073ecd

github-actions bot reviewed Dec 14, 2024

View reviewed changes

jumerckx added 2 commits December 14, 2024 20:31

other way 'round

585c485

formatting

0c56d35

wsmoses merged commit 65e9976 into main Dec 14, 2024
22 of 46 checks passed

wsmoses deleted the interp2 branch December 14, 2024 20:05

		(TracedUtils.promote_to(TracedRNumber{Int}, i isa Colon ? 1 : first(i)) - 1).mlir_data for
		i in indices

-TracedUtils.promote_to(::TracedRArray{T,N}, rhs) where {T,N} = TracedUtils.promote_to(TracedRArray{T,N}, rhs)
+function TracedUtils.promote_to(::TracedRArray{T,N}, rhs) where {T,N}
+    return TracedUtils.promote_to(TracedRArray{T,N}, rhs)
+end

-Base.float(x::TracedRNumber{T}) where {T} = TracedUtils.promote_to(TracedRNumber{float(T)}, x)
+function Base.float(x::TracedRNumber{T}) where {T}
+    return TracedUtils.promote_to(TracedRNumber{float(T)}, x)
+end


		struct TypeCast{T<:ReactantPrimitive} <: Function end

		(::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2} = TracedUtils.promote_to(TracedRNumber{T}, x)

-(::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2} = TracedUtils.promote_to(TracedRNumber{T}, x)
+function (::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2}
+    return TracedUtils.promote_to(TracedRNumber{T}, x)
+end

	res isa TracedRNumber && (res = Reactant.TracedUtils.broadcast_to_size(res, (1,)))
	res isa TracedRNumber &&
	(res = Reactant.TracedUtils.broadcast_to_size(res, (1,)))

	Ops.iota(Int64, [size(X)...]; iota_dimension=2), TracedUtils.broadcast_to_size(k, size(X))
	Ops.iota(Int64, [size(X)...]; iota_dimension=2),
	TracedUtils.broadcast_to_size(k, size(X)),

	push!(
	overdubbed_code,
	Core.ReturnNode(Core.SSAValue(length(overdubbed_code)))
	)

	push!(overdubbed_code,
	Expr(:call,
	make_oc,
	octup,
	rt,
	src,
	ocnargs,
	ocva,
	farg
	)
	)

		mi = ccall(:jl_specializations_get_linfo, Ref{Core.MethodInstance},
		(Any, Any, Any), match.method, match.spec_types, match.sparams)


	push!(overdubbed_code, :(2:$expr_lastindex))
	push!(overdubbed_code, :(2:($expr_lastindex)))

Interp2 #365

Interp2 #365

Conversation

wsmoses commented Dec 11, 2024

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot left a comment

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot left a comment

Choose a reason for hiding this comment

github-actions bot left a comment

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment

github-actions bot Dec 14, 2024

Choose a reason for hiding this comment