Modifications to "get_working_device" function in Pytorch. Added the capability to set a Pytorch device using the "set_device" function#891

Merged

lior-dikstein merged 5 commits intomainfrom pytorch_device

Dec 31, 2023

+184-29

Commits on Dec 31, 2023

Modifications to "get_working_device" function in Pytorch. Added the capability to set a Pytorch device using the "set_device" function. Implemented using DeviceManager with Singleton Pattern.
liord
committed
Modifications to "get_working_device" function in Pytorch. Added the capability to set a Pytorch device using the "set_device" function. Implemented using DeviceManager with Singleton Pattern.
liord
committed
Added print device to device test
liord
committed
Delete print and fix test
liord
committed
Add type hints and fix comments
liord
committed