ruda-tensor-device 0.21.12

Ruda device tensor backend and domain-library dispatch.
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

use ruda_tensor::{
    backend::ExecutionError,
    ops::{TransactionOps, TransactionPrimitive, TransactionPrimitiveData},
};

use crate::{DeviceBackend, DeviceRuntime, FloatElement, IntElement, element::BoolElement};

impl<R, F, I, BT> TransactionOps<Self> for DeviceBackend<R, F, I, BT>
where
    R: DeviceRuntime,
    F: FloatElement,
    I: IntElement,
    BT: BoolElement,
{
    async fn tr_execute(
        transaction: TransactionPrimitive<Self>,
    ) -> Result<TransactionPrimitiveData, ExecutionError> {
        ruda_kernel::tensor::transaction::execute(
            ruda_kernel::tensor::transaction::ReadbackBatch {
                read_floats: transaction.read_floats,
                read_qfloats: transaction.read_qfloats,
                read_ints: transaction.read_ints,
                read_bools: transaction.read_bools,
            },
        ).await
    }
}