huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

All tests are panicking instead of random failure.

Browse Source
This commit is contained in:
Nicolas Patry
2023-11-11 17:06:35 +01:00
parent 54355ff997
commit 3900091e75
7 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/src/metal_backend.rs
View File

@ -172,8 +172,8 @@ impl BackendStorage for MetalStorage {
.unwrap();
} else {
let name = match self.dtype {
DType::F32 => "affine_float",
DType::F16 => "affine_half",
DType::F32 => "affine_float_strided",
DType::F16 => "affine_half_strided",
dtype => todo!("Affine {dtype:?}"),
};
candle_metal_kernels::call_affine_strided(
@ -829,7 +829,6 @@ impl BackendStorage for MetalStorage {
fn copy_strided_src(&self, dst: &mut Self, dst_offset: usize, src_l: &Layout) -> Result<()> {
let src_shape = src_l.shape();
let el_count = src_shape.elem_count();
// todo!("COPY STRIDED {src_shape:?} {el_count} {src_l:?} {dst_offset}");
if el_count == 0 {
return Ok(());
}
@ -851,7 +850,7 @@ impl BackendStorage for MetalStorage {
&src_l.stride(),
src_l.start_offset() * self.dtype.size_in_bytes(),
&mut dst.buffer,
dst_offset,
dst_offset * dst.dtype.size_in_bytes(),
)
.map_err(MetalError::from)?;
// command_buffer.commit();