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

fmt

Browse Source
This commit is contained in:
Ivar Flakstad
2024-01-22 21:23:44 +01:00
parent 086b6ef6b6
commit 077e781f53
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-metal-kernels/src/tests.rs
View File

@ -633,7 +633,7 @@ fn run_reduce<T: Clone>(v: &[T], out_length: usize, name: &'static str) -> Vec<T
let output = device.new_buffer((out_length * core::mem::size_of::<T>()) as u64, options); let output = device.new_buffer((out_length * core::mem::size_of::<T>()) as u64, options);
let dims = vec![v.len()]; let dims = vec![v.len()];
let strides = vec![1]; let strides = vec![1];
let result = call_reduce_strided( call_reduce_strided(
&device, &device,
command_buffer, command_buffer,
&kernels, &kernels,
@ -644,7 +644,8 @@ fn run_reduce<T: Clone>(v: &[T], out_length: usize, name: &'static str) -> Vec<T
&input, &input,
0, 0,
&output, &output,
).unwrap(); )
.unwrap();
command_buffer.commit(); command_buffer.commit();
command_buffer.wait_until_completed(); command_buffer.wait_until_completed();