diff --git a/ext/AMDGPUExt/AMDGPUExt.jl b/ext/AMDGPUExt/AMDGPUExt.jl index 1e948048..fc8e79a6 100644 --- a/ext/AMDGPUExt/AMDGPUExt.jl +++ b/ext/AMDGPUExt/AMDGPUExt.jl @@ -2,7 +2,7 @@ module AMDGPUExt using AMDGPU, AMDGPU.ROCKernels -import FastIce.Architecture: heuristic_groupsize, set_device!, get_device +import FastIce.Architectures: heuristic_groupsize, set_device!, get_device set_device!(dev::HIPDevice) = AMDGPU.device!(dev) diff --git a/ext/CUDAExt/CUDAExt.jl b/ext/CUDAExt/CUDAExt.jl index 919fef7a..b6653811 100644 --- a/ext/CUDAExt/CUDAExt.jl +++ b/ext/CUDAExt/CUDAExt.jl @@ -2,7 +2,7 @@ module CUDAExt using CUDA, CUDA.CUDAKernels -import FastIce.Architecture: heuristic_groupsize, set_device!, get_device +import FastIce.Architectures: heuristic_groupsize, set_device!, get_device set_device!(dev::CuDevice) = CUDA.device!(dev)