Module device

Module device 

Source
Expand description

Device selection (CPU / Metal / CUDA).

Functionsยง

select_device
Selects the compute device based on enabled features (falls back to CPU).