sp1_gpu_sys/
reduce.rs

1use crate::runtime::KernelPtr;
2
3extern "C" {
4    pub fn koala_bear_sum_block_reduce_kernel() -> KernelPtr;
5    pub fn koala_bear_sum_partial_block_reduce_kernel() -> KernelPtr;
6    pub fn koala_bear_extension_sum_block_reduce_kernel() -> KernelPtr;
7    pub fn koala_bear_extension_sum_partial_block_reduce_kernel() -> KernelPtr;
8
9    pub fn partial_inner_product_koala_bear_kernel() -> KernelPtr;
10    pub fn partial_inner_product_koala_bear_extension_kernel() -> KernelPtr;
11    pub fn partial_inner_product_koala_bear_base_extension_kernel() -> KernelPtr;
12
13    pub fn partial_dot_koala_bear_kernel() -> KernelPtr;
14    pub fn partial_dot_koala_bear_extension_kernel() -> KernelPtr;
15    pub fn partial_dot_koala_bear_base_extension_kernel() -> KernelPtr;
16
17    pub fn dot_along_short_dimension_kernel_koala_bear_base_base() -> KernelPtr;
18    pub fn dot_along_short_dimension_kernel_koala_bear_base_extension() -> KernelPtr;
19    pub fn dot_along_short_dimension_kernel_koala_bear_extension_extension() -> KernelPtr;
20}
sp1_gpu_sys/reduce.rs

sp1_gpu_sys/
reduce.rs
