cubecl-core 0.10.0-pre.3

CubeCL core create
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use crate::prelude::*;

/// Bulk copy `length` elements between two array-likes without intermediates.
///
/// # Arguments
///
/// `from` - The array/tensor/shared memory to copy from
/// `from_index` - The `from` index to start the copy from
/// `to` - The array/tensor/shared memory to copy to
/// `to_index` - The `to` index to copy the elements to
///
/// # Example
///
/// ```ignore
/// copy_bulk(input.as_slice(), idx, shared, shared_idx, 16);
/// ```
pub fn copy_bulk<C: CubePrimitive>(
    _from: &Slice<C>,
    _from_index: usize,
    _to: &mut SliceMut<C>,
    _to_index: usize,
    _length: usize,
) {
}

pub mod copy_bulk {
    use crate::ir::{CopyMemoryBulkOperator, Instruction, Operator, Scope};

    use super::*;

    /// The expand function for [`copy_bulk()`]
    pub fn expand<C: CubePrimitive>(
        scope: &mut Scope,
        from: SliceExpand<C, ReadOnly>,
        from_index: NativeExpand<usize>,
        to: SliceExpand<C, ReadWrite>,
        to_index: NativeExpand<usize>,
        length: usize,
    ) {
        let (input, input_offset) = from.__to_raw_parts();
        let (to, to_offset) = to.__to_raw_parts();

        scope.register(Instruction::new(
            Operator::CopyMemoryBulk(CopyMemoryBulkOperator {
                out_index: to_index.expand.consume(),
                input,
                in_index: from_index.expand.consume(),
                len: length,
                offset_input: input_offset,
                offset_out: to_offset,
            }),
            to,
        ));
    }
}