missing Ext

EnzymeAD · Dec 16, 2024 · 50d7e8b · 50d7e8b
1 parent 520f71f
commit 50d7e8b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 23 deletions.
diff --git a/Project.toml b/Project.toml
@@ -32,7 +32,7 @@ path = "lib/ReactantCore"
 ReactantAbstractFFTsExt = "AbstractFFTs"
 ReactantArrayInterfaceExt = "ArrayInterface"
 ReactantNNlibExt = "NNlib"
-ReactantSpecialFunctions = "SpecialFunctions"
+ReactantSpecialFunctionsExt = "SpecialFunctions"
 ReactantStatisticsExt = "Statistics"
 ReactantYaoBlocksExt = "YaoBlocks"
 

diff --git a/src/Reactant.jl b/src/Reactant.jl
@@ -16,25 +16,12 @@ using Enzyme
 struct ReactantABI <: Enzyme.EnzymeCore.ABI end
 
 @static if isdefined(Core, :BFloat16)
-    const ReactantPrimitive = Union{
-        Bool,
-        Int8,
-        UInt8,
-        Int16,
-        UInt16,
-        Int32,
-        UInt32,
-        Int64,
-        UInt64,
-        Float16,
-        Core.BFloat16,
-        Float32,
-        Float64,
-        Complex{Float32},
-        Complex{Float64},
-    }
+    ReactantFloat = Union{Float16, BFloat16, Float32, Float64}
 else
-    const ReactantPrimitive = Union{
+    ReactantFloat = Union{Float16, Float32, Float64}
+end
+
+const ReactantPrimitive = Union{
         Bool,
         Int8,
         UInt8,
@@ -44,13 +31,10 @@ else
         UInt32,
         Int64,
         UInt64,
-        Float16,
-        Float32,
-        Float64,
         Complex{Float32},
         Complex{Float64},
+        ReactantFloat...
     }
-end
 
 abstract type RArray{T<:ReactantPrimitive,N} <: AbstractArray{T,N} end
 abstract type RNumber{T<:ReactantPrimitive} <: Number end