Skip to main content

default_device

Function default_device 

Source
pub fn default_device() -> &'static Device
Expand description

Return the device docbert-plaid uses for all tensor compute.

Cached on first call. The choice is:

  • With the cuda feature and a usable CUDA device → Device::Cuda(0).
  • Otherwise → Device::Cpu.