Skip to main content

ruda_tensor/device/
adapters.rs

1#[cfg(feature = "wgpu")]
2mod wgpu {
3    use crate::device::DeviceOps;
4    use ruda_driver_wgpu::WgpuDevice;
5
6    impl DeviceOps for WgpuDevice {}
7}
8
9#[cfg(feature = "cuda")]
10mod cuda {
11    use crate::device::DeviceOps;
12    use ruda_driver_cuda::CudaDevice;
13
14    impl DeviceOps for CudaDevice {}
15}
16
17#[cfg(feature = "cpu")]
18mod cpu {
19    use crate::device::DeviceOps;
20    use ruda_driver_cpu::CpuDevice;
21
22    impl DeviceOps for CpuDevice {}
23}
24
25#[cfg(feature = "hip")]
26mod hip {
27    use crate::device::DeviceOps;
28    use ruda_driver_hip::AmdDevice;
29
30    impl DeviceOps for AmdDevice {}
31}