More bfloat fix (#1728)

* More bfloat fix * Update bfloat16s.jl * Update bfloat16s.jl
EnzymeAD · Aug 12, 2024 · 0cf47c1 · 0cf47c1
1 parent cf619b3
commit 0cf47c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/ext/EnzymeBFloat16sExt.jl b/ext/EnzymeBFloat16sExt.jl
@@ -4,7 +4,7 @@ using BFloat16s
 using Enzyme
 
 function Enzyme.typetree_inner(::Type{BFloat16}, ctx, dl, seen::Enzyme.Compiler.TypeTreeTable)
-    return TypeTree(Enzyme.API.DT_BFloat16, -1, ctx)
+    return Enzyme.TypeTree(Enzyme.API.DT_BFloat16, -1, ctx)
 end
 
 end
diff --git a/src/typetree.jl b/src/typetree.jl
@@ -121,11 +121,6 @@ function typetree_inner(::Type{BigFloat}, ctx, dl, seen::TypeTreeTable)
     return TypeTree()
 end
 
-function typetree_inner(::Type{T}, ctx, dl, seen::TypeTreeTable) where {T<:AbstractFloat}
-    GPUCompiler.@safe_warn "Unknown floating point type" T
-    return TypeTree()
-end
-
 function typetree_inner(::Type{<:DataType}, ctx, dl, seen::TypeTreeTable)
     return TypeTree()
 end
@@ -225,6 +220,10 @@ function typetree_inner(@nospecialize(T), ctx, dl, seen::TypeTreeTable)
         end
     end
 
+    if T <: AbstractFloat
+        throw(AssertionError("Unknown floating point type $T"))
+    end
+
     try
         fieldcount(T)
     catch