From ebebf6fe4c71ccd5c9dd49acc9d5fa827fa0c4fc Mon Sep 17 00:00:00 2001 From: Ludovic Raess Date: Mon, 8 Apr 2024 20:04:02 +0200 Subject: [PATCH] Add method --- src/Architectures.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/Architectures.jl b/src/Architectures.jl index 40f3dd12..e4f717ce 100644 --- a/src/Architectures.jl +++ b/src/Architectures.jl @@ -27,6 +27,8 @@ struct SingleDeviceArchitecture{B,D} <: Architecture end end +SingleDeviceArchitecture(arch::Architecture) = SingleDeviceArchitecture(get_backend(arch), get_device(arch)) + """ Arch(backend::Backend; device_id::Integer=1)