Skip to content

Commit

Permalink
fix: oneAPI _get_device
Browse files Browse the repository at this point in the history
  • Loading branch information
@avik-pal
avik-pal committed Jul 12, 2024
1 parent 0b303f9 commit e2c44fe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ext/LuxDeviceUtilsoneAPIExt.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit e2c44fe

Please sign in to comment.