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

Where cond get_strided_index conditionally based on function constants

Browse Source
This commit is contained in:
Ivar Flakstad
2024-01-22 20:59:02 +01:00
parent fd7c856564
commit 933716b374
4 changed files with 41 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -822,10 +822,13 @@ impl BackendStorage for MetalStorage {
layout.stride(),
layout.start_offset() * self.dtype.size_in_bytes(),
),
!layout.is_contiguous(),
&t.buffer,
(&t_l.stride(), t_l.start_offset() * t.dtype.size_in_bytes()),
!t_l.is_contiguous(),
&f.buffer,
(&f_l.stride(), f_l.start_offset() * f.dtype.size_in_bytes()),
!f_l.is_contiguous(),
&buffer,
)
.map_err(MetalError::from)?;