gllm_kernels/
device.rs

1//! Device helpers using Burn backends.
2
3use burn::tensor::backend::Backend;
4
5use crate::backend::{select_device, DefaultBackend};
6
7/// Default device for the configured backend.
8pub type DefaultDevice = <DefaultBackend as Backend>::Device;
9
10/// Select the default device for the configured backend.
11pub fn default_device() -> DefaultDevice {
12    select_device::<DefaultBackend>()
13}
14
15/// Select the default device for a specific backend.
16pub fn device_for<B: Backend>() -> B::Device {
17    select_device::<B>()
18}