From c63ecfb7de4973a777735fe5eedee786853ebed8 Mon Sep 17 00:00:00 2001 From: William Moses Date: Mon, 23 Dec 2024 15:19:20 -0500 Subject: [PATCH] Update EnzymeGPUArraysCoreExt.jl --- ext/EnzymeGPUArraysCoreExt.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)