fix: import ordering

EnzymeAD · Oct 5, 2024 · 40af781 · 40af781
1 parent 36a6fea
commit 40af781
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/src/Reactant.jl b/src/Reactant.jl
@@ -65,8 +65,8 @@ include("XLA.jl")
 include("Interpreter.jl")
 include("utils.jl")
 include("ConcreteRArray.jl")
-include("TracedRArray.jl")
 include("TracedRNumber.jl")
+include("TracedRArray.jl")
 include("Tracing.jl")
 include("Compiler.jl")
 

diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
@@ -213,10 +213,10 @@ end
 
 promote_to(::TracedRArray{T,N}, rhs) where {T,N} = promote_to(TracedRArray{T,N}, rhs)
 
-struct TypeCast{T<:Number} <: Function end
-
-elem_apply(::Type{T}, x::TracedRArray{T}) where {T<:Number} = x
-function elem_apply(::Type{T}, x::TracedRArray{T2}) where {T<:Number,T2<:Number}
+elem_apply(::Type{T}, x::TracedRArray{T}) where {T<:ReactantPrimitives} = x
+function elem_apply(
+    ::Type{T}, x::TracedRArray{T2}
+) where {T<:ReactantPrimitives,T2<:ReactantPrimitives}
     # Special Path to prevent going down a despecialized path
     return elem_apply(TypeCast{T}(), x)
 end

diff --git a/src/TracedRNumber.jl b/src/TracedRNumber.jl
@@ -158,6 +158,8 @@ for (jlop, hloop) in (
     end
 end
 
+struct TypeCast{T<:ReactantPrimitives} <: Function end
+
 (::TypeCast{T})(x::TracedRNumber{T2}) where {T,T2} = promote_to(TracedRNumber{T}, x)
 
 Base.float(x::TracedRNumber{T}) where {T} = promote_to(TracedRNumber{float(T)}, x)