Remove vararg tuple warning (EnzymeAD#1395)

MilesCranmer · Apr 20, 2024 · 4230cc0 · 4230cc0
1 parent 43340b3
commit 4230cc0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -461,8 +461,6 @@ end
         end
     end
 
-    @inline is_concrete_tuple(x::T2) where T2 = (x <: Tuple) && !(x === Tuple) && !(x isa UnionAll)
-
     @assert !Base.isabstracttype(T)
     if !(Base.isconcretetype(T) || is_concrete_tuple(T) || T isa UnionAll)
         throw(AssertionError("Type $T is not concrete type or concrete tuple"))

diff --git a/src/typetree.jl b/src/typetree.jl
@@ -191,6 +191,12 @@ function typetree_inner(@nospecialize(T), ctx, dl, seen::TypeTreeTable)
         return TypeTree()
     end
 
+    @static if VERSION >= v"1.7.0"
+        if is_concrete_tuple(T) && any(T2 isa Core.TypeofVararg for T2 in T.parameters)
+            return TypeTree()
+        end
+    end
+
     try
         fieldcount(T)
     catch

diff --git a/src/utils.jl b/src/utils.jl
@@ -1,6 +1,9 @@
 unsafe_to_pointer(ptr) = ccall(Base.@cfunction(x->x, Ptr{Cvoid}, (Ptr{Cvoid},)), Ptr{Cvoid}, (Any,), ptr)
 export unsafe_to_pointer
 
+@inline is_concrete_tuple(x::T2) where T2 = (x <: Tuple) && !(x === Tuple) && !(x isa UnionAll)
+export is_concrete_tuple
+
 const Tracked = 10
 const Derived = 11
 export Tracked, Derived