diff --git a/ext/EnzymeGPUArraysCoreExt.jl b/ext/EnzymeGPUArraysCoreExt.jl index 9f1edbf8fd..94b377700e 100644 --- a/ext/EnzymeGPUArraysCoreExt.jl +++ b/ext/EnzymeGPUArraysCoreExt.jl @@ -4,11 +4,11 @@ using GPUArraysCore using Enzyme @inline function Enzyme.onehot(x::AbstractGPUArray) - Enzyme.onehot_internal(zerosetfn, x, 0, length(x)) + Enzyme.onehot_internal(Enzyme.zerosetfn, x, 0, length(x)) end @inline function Enzyme.onehot(x::AbstractGPUArray, start::Int, endl::Int) - Enzyme.onehot_internal(zerosetfn, x, start-1, endl-start+1) + Enzyme.onehot_internal(Enzyme.zerosetfn, x, start-1, endl-start+1) end function Enzyme.zerosetfn(x::AbstractGPUArray, i::Int)