sp1-gpu-sys 6.1.0

FFI bindings and CUDA build system for SP1-GPU
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::runtime::KernelPtr;

extern "C" {
    pub fn koala_bear_sum_block_reduce_kernel() -> KernelPtr;
    pub fn koala_bear_sum_partial_block_reduce_kernel() -> KernelPtr;
    pub fn koala_bear_extension_sum_block_reduce_kernel() -> KernelPtr;
    pub fn koala_bear_extension_sum_partial_block_reduce_kernel() -> KernelPtr;

    pub fn partial_inner_product_koala_bear_kernel() -> KernelPtr;
    pub fn partial_inner_product_koala_bear_extension_kernel() -> KernelPtr;
    pub fn partial_inner_product_koala_bear_base_extension_kernel() -> KernelPtr;

    pub fn partial_dot_koala_bear_kernel() -> KernelPtr;
    pub fn partial_dot_koala_bear_extension_kernel() -> KernelPtr;
    pub fn partial_dot_koala_bear_base_extension_kernel() -> KernelPtr;

    pub fn dot_along_short_dimension_kernel_koala_bear_base_base() -> KernelPtr;
    pub fn dot_along_short_dimension_kernel_koala_bear_base_extension() -> KernelPtr;
    pub fn dot_along_short_dimension_kernel_koala_bear_extension_extension() -> KernelPtr;
}