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

Fix size-in-bytes for u8. (#351)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-08 22:15:18 +02:00
committed by GitHub
parent 608b2358c6
commit b9864e1357
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/dtype.rs
View File

@ -43,7 +43,7 @@ impl DType {
pub fn size_in_bytes(&self) -> usize { pub fn size_in_bytes(&self) -> usize {
match self { match self {
Self::U8 => 4, Self::U8 => 1,
Self::U32 => 4, Self::U32 => 4,
Self::BF16 => 2, Self::BF16 => 2,
Self::F16 => 2, Self::F16 => 2,