From e2c44fe10fb5cf8191f38feadb671fb3a5da2f7d Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Fri, 12 Jul 2024 14:57:14 -0700 Subject: [PATCH] fix: oneAPI _get_device --- ext/LuxDeviceUtilsoneAPIExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/LuxDeviceUtilsoneAPIExt.jl b/ext/LuxDeviceUtilsoneAPIExt.jl index 00e73e6..f9da407 100644 --- a/ext/LuxDeviceUtilsoneAPIExt.jl +++ b/ext/LuxDeviceUtilsoneAPIExt.jl @@ -25,7 +25,7 @@ end LuxDeviceUtils.default_device_rng(::LuxoneAPIDevice) = GPUArrays.default_rng(oneArray) # Query Device from Array -LuxDeviceUtils.get_device(::oneArray) = LuxoneAPIDevice() +LuxDeviceUtils._get_device(::oneArray) = LuxoneAPIDevice() LuxDeviceUtils._get_device_type(::oneArray) = LuxoneAPIDevice