Skip to content

Commit

Permalink
Update docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
luraess committed Apr 8, 2024
1 parent d662572 commit f518cc6
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/Architectures.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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))

"""
Expand Down

0 comments on commit f518cc6

Please sign in to comment.