huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add an easy way to create tensor objects.

Browse Source
This commit is contained in:
laurent
2023-06-19 20:59:26 +01:00
parent 01eeb0e72f
commit 26d6288eb6
6 changed files with 77 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/dtype.rs
View File

@ -1,3 +1,5 @@
use crate::CpuStorage;
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
pub enum DType {
F32,
@ -13,14 +15,24 @@ impl DType {
}
}
pub trait WithDType {
pub trait WithDType: Sized + Copy {
const DTYPE: DType;
fn to_cpu_storage(data: &[Self]) -> CpuStorage;
}
impl WithDType for f32 {
const DTYPE: DType = DType::F32;
fn to_cpu_storage(data: &[Self]) -> CpuStorage {
CpuStorage::F32(data.to_vec())
}
}
impl WithDType for f64 {
const DTYPE: DType = DType::F64;
fn to_cpu_storage(data: &[Self]) -> CpuStorage {
CpuStorage::F64(data.to_vec())
}
}