diff --git a/src/Architectures.jl b/src/Architectures.jl index e4f717ce..2765ec75 100644 --- a/src/Architectures.jl +++ b/src/Architectures.jl @@ -27,6 +27,11 @@ struct SingleDeviceArchitecture{B,D} <: Architecture end end +""" + SingleDeviceArchitecture(arch::Architecture) + +Create a `SingleDeviceArchitecture` object retrieving backend and device from `arch`. +""" SingleDeviceArchitecture(arch::Architecture) = SingleDeviceArchitecture(get_backend(arch), get_device(arch)) """