sp1-gpu-cudart 6.2.2

High-level Rust API for SP1 GPU operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

// Async CopyIntoBackend/CopyToBackend impls removed - use sync DeviceMle/DeviceTensor methods instead

use crate::{sync::CudaSend, TaskScope};
use slop_multilinear::{Mle, MleEval, Point};

impl<T> CudaSend for Mle<T, TaskScope> {
    #[inline]
    unsafe fn send_to_scope(self, scope: &TaskScope) -> Self {
        let guts = self.into_guts().send_to_scope(scope);
        Mle::new(guts)
    }
}

impl<T> CudaSend for MleEval<T, TaskScope> {
    #[inline]
    unsafe fn send_to_scope(self, scope: &TaskScope) -> Self {
        let evaluations = self.into_evaluations().send_to_scope(scope);
        MleEval::new(evaluations)
    }
}

impl<T> CudaSend for Point<T, TaskScope> {
    #[inline]
    unsafe fn send_to_scope(self, scope: &TaskScope) -> Self {
        let values = self.into_values().send_to_scope(scope);
        Point::new(values)
    }
}