fix primitive_type for complex (#193)

* fix primitive_type for complex

* add simple complex runtime test

Author: Pangoraw

src/XLA.jl

@@ -233,8 +233,8 @@ end
 
 @inline primitive_type(::Type{Float64}) = 12
 
-@inline primitive_type(::Type{Complex{Float32}}) = 24
-@inline primitive_type(::Type{Complex{Float64}}) = 25
+@inline primitive_type(::Type{Complex{Float32}}) = 15
+@inline primitive_type(::Type{Complex{Float64}}) = 18
 
 function ArrayFromHostBuffer(client::Client, array::Array{T,N}, device) where {T,N}
     sizear = Int64[s for s in reverse(size(array))]

test/basic.jl

@@ -438,3 +438,10 @@ end
     @test size(f(y)) == size(x)
     @test eltype(f(y)) == eltype(x)
 end
+
+@testset "Complex runtime: $CT" for CT in (ComplexF32, ComplexF64)
+    a = Reactant.to_rarray(ones(CT, 2))
+    b = Reactant.to_rarray(ones(CT, 2))
+    c = Reactant.compile(+, (a, b))(a, b)
+    @test c == ones(CT, 2) + ones(CT, 2)
+end