torch.cpu#
Created On: Jul 11, 2023 | Last Updated: Oct 11, 2023
This package implements abstractions found in torch.cuda
to facilitate writing device-agnostic code.
current_device |
Returns current device for cpu. |
current_stream |
Returns the currently selected |
is_available |
Returns a bool indicating if CPU is currently available. |
synchronize |
Waits for all kernels in all streams on the CPU device to complete. |
stream |
Wrapper around the Context-manager StreamContext that selects a given stream. |
set_device |
Sets the current device, in CPU we do nothing. |
device_count |
Returns number of CPU devices (not cores). |
StreamContext |
Context-manager that selects a given stream. |