Skip to content

Commit da94cf8

Browse files
committed
Generalize KA.get_backend to AnyConcreteRArray
1 parent 747d88f commit da94cf8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/ReactantKernelAbstractionsExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ function KA.ones(b::ReactantBackend, ::Type{T}, dims::Tuple) where {T}
4141
end
4242

4343
KA.get_backend(::Reactant.AnyTracedRArray) = ReactantBackend()
44-
KA.get_backend(::Reactant.AnyConcretePJRTArray) = ReactantBackend()
44+
KA.get_backend(::Reactant.AnyConcreteRArray) = ReactantBackend()
4545
function KA.synchronize(::ReactantBackend) end
4646

4747
Adapt.adapt_storage(::ReactantBackend, a::Array) = a

0 commit comments

Comments
 (0)