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

Get the cpu tests to run.

Browse Source
This commit is contained in:
laurent
2023-06-28 14:32:02 +01:00
parent 14449ff80c
commit caafef6cc1
5 changed files with 6 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/cpu_backend.rs
View File

@ -229,10 +229,6 @@ impl CpuStorage {
D::cpu_storage_as_slice(self)
}
pub fn as_mut_slice<D: crate::WithDType>(&mut self) -> Result<&mut [D]> {
D::cpu_storage_as_mut_slice(self)
}
pub(crate) fn to_dtype(&self, layout: &Layout, dtype: DType) -> Result<Self> {
// TODO: find a way around the quadratic number of cases below.
match (self, dtype) {
@ -581,6 +577,7 @@ impl CpuStorage {
layout_f: &Layout,
) -> Result<Self> {
// TODO: Support types that could be casted to a boolean.
// TODO: this should use the layout.
let pred = self.as_slice::<u32>()?;
match (t, f) {
(Self::BF16(t), Self::BF16(f)) => {
@ -618,6 +615,7 @@ impl CpuStorage {
hidden_size: usize,
vocab_size: usize,
) -> Result<Self> {
// TODO: this should use the layout.
let ids = self.as_slice::<u32>()?;
map1!(vs, take_impl1, ids, layout, vocab_size, hidden_size)
}