Skip to content

Commit 10456f9

Browse files
committed
fix(cpu): correct module path for masked_count_kernel
Update masked_count_kernel call to use correct module path kernels::index::masked_count_kernel for non-x86_64 architectures.
1 parent 72dcef8 commit 10456f9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/runtime/cpu/helpers/indexing.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -501,7 +501,7 @@ pub fn masked_select_impl(
501501
#[cfg(not(target_arch = "x86_64"))]
502502
{
503503
// Count true elements first
504-
let count = unsafe { kernels::masked_count_kernel(mask_ptr as *const u8, numel) };
504+
let count = unsafe { kernels::index::masked_count_kernel(mask_ptr as *const u8, numel) };
505505

506506
// Allocate output with correct size
507507
let out = Tensor::<CpuRuntime>::empty(&[count], dtype, &client.device);

0 commit comments

Comments
 (0)