Update EnzymeGPUArraysCoreExt.jl

EnzymeAD · Dec 23, 2024 · c63ecfb · c63ecfb
1 parent 86e86e9
commit c63ecfb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)