We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 4b771fe + d903553 commit b0356a2Copy full SHA for b0356a2
NDTensors/ext/NDTensorsAMDGPUExt/adapt.jl
@@ -11,7 +11,7 @@ using Adapt: Adapt, adapt
11
using AMDGPU: AMDGPU, ROCArray, ROCVector
12
using Functors: fmap
13
14
-function AMDGPUExtensions.roc(xs, storagemode=default_type_parameter(ROCArray, storagemode))
+function AMDGPUExtensions.roc(xs; storagemode=default_type_parameter(ROCArray, storagemode))
15
return fmap(x -> adapt(ROCArrayAdaptor{storagemode}(), x), xs)
16
end
17
0 commit comments