Add isless (#923)

* Add isless

* Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* fix test

* Upgrade Reactant_jll to v0.0.89

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Mosè Giordano <[email protected]>

Author: 3 people

Project.toml

@@ -86,7 +86,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.5"
-Reactant_jll = "0.0.88"
+Reactant_jll = "0.0.89"
 Scratch = "1.2"
 Sockets = "1.10"
 SpecialFunctions = "2.4"

ext/ReactantCUDAExt.jl

@@ -50,6 +50,10 @@ function Base.convert(::Type{T}, RN::CuTracedRNumber) where {T<:Number}
     return Base.convert(T, Base.getindex(RN))
 end
 
+Base.isless(a::CuTracedRNumber, b::CuTracedRNumber) = Base.isless(a[], b[])
+Base.isless(a, b::CuTracedRNumber) = Base.isless(a, b[])
+Base.isless(a::CuTracedRNumber, b) = Base.isless(a[], b)
+
 function Base.promote_rule(
     ::Type{<:CuTracedRNumber{T}}, ::Type{<:CuTracedRNumber{T2}}
 ) where {T,T2}

test/integration/cuda.jl

@@ -154,3 +154,24 @@ end
     @jit mul_number!(A, B)
     @test all(Array(A) .≈ oA .* 3.1)
 end
+
+function searchsorted_kernel!(x, y)
+    i = threadIdx().x
+    times = 0:0.01:4.5
+    z = searchsortedfirst(times, y)
+    x[i] = z
+    return nothing
+end
+
+function searchsorted!(x, y)
+    @cuda blocks = 1 threads = length(x) searchsorted_kernel!(x, y)
+    return nothing
+end
+
+@testset "Search sorted" begin
+    oA = collect(Float64, 1:1:64)
+    A = Reactant.to_rarray(oA)
+    B = ConcreteRNumber(3.1)
+    @jit searchsorted!(A, B)
+    @test all(Array(A) .≈ 311)
+end