mirror of
https://github.com/huggingface/candle.git
synced 2025-06-21 12:20:46 +00:00
135 lines
3.4 KiB
Rust
135 lines
3.4 KiB
Rust
#![allow(dead_code)]
|
|
use crate::{CpuStorage, DType, Error, Layout, Result, Shape};
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct CudaDevice;
|
|
|
|
#[derive(Debug)]
|
|
pub struct CudaStorage;
|
|
|
|
macro_rules! fail {
|
|
() => {
|
|
unimplemented!("cuda support has not been enabled")
|
|
};
|
|
}
|
|
|
|
impl crate::backend::BackendStorage for CudaStorage {
|
|
type Device = CudaDevice;
|
|
|
|
fn try_clone(&self, _: &Layout) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn dtype(&self) -> DType {
|
|
fail!()
|
|
}
|
|
|
|
fn device(&self) -> &Self::Device {
|
|
fail!()
|
|
}
|
|
|
|
fn to_cpu_storage(&self) -> Result<CpuStorage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn affine(&self, _: &Layout, _: f64, _: f64) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn elu(&self, _: &Layout, _: f64) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn sum(&self, _: &Layout, _: &[usize]) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn divide_by_sum_over_dim(&mut self, _: &Shape, _: usize) -> Result<()> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn to_dtype(&self, _: &Layout, _: DType) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn unary_impl<B: crate::op::UnaryOp>(&self, _: &Layout) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn binary_impl<B: crate::op::BinaryOp>(
|
|
&self,
|
|
_: &Self,
|
|
_: &Layout,
|
|
_: &Layout,
|
|
) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn where_cond(&self, _: &Layout, _: &Self, _: &Layout, _: &Self, _: &Layout) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn conv1d(
|
|
&self,
|
|
_: &Layout,
|
|
_: &Self,
|
|
_: &Layout,
|
|
_: &crate::conv::ParamsConv1D,
|
|
) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn embedding(&self, _: &Layout, _: &Self, _: &Layout) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn matmul(
|
|
&self,
|
|
_: &Self,
|
|
_: (usize, usize, usize, usize),
|
|
_: &Layout,
|
|
_: &Layout,
|
|
) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn copy_strided_src(&self, _: &mut Self, _: usize, _: &Layout) -> Result<()> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
}
|
|
|
|
impl crate::backend::BackendDevice for CudaDevice {
|
|
type Storage = CudaStorage;
|
|
fn new(_: usize) -> Result<Self> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn location(&self) -> crate::DeviceLocation {
|
|
fail!()
|
|
}
|
|
|
|
fn same_device(&self, _: &Self) -> bool {
|
|
fail!()
|
|
}
|
|
|
|
fn zeros_impl(&self, _shape: &Shape, _dtype: DType) -> Result<Self::Storage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn ones_impl(&self, _shape: &Shape, _dtype: DType) -> Result<Self::Storage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn storage_from_cpu_storage(&self, _: &CpuStorage) -> Result<Self::Storage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn rand_uniform(&self, _: &Shape, _: DType, _: f64, _: f64) -> Result<Self::Storage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
|
|
fn rand_normal(&self, _: &Shape, _: DType, _: f64, _: f64) -> Result<Self::Storage> {
|
|
Err(Error::NotCompiledWithCudaSupport)
|
|
}
|
|
}
|