cubecl-core 0.11.0-pre.1

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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::prelude::*;

/// Bulk copy `length` elements between two array-likes without intermediates.
///
/// # Arguments
///
/// `from` - slice to copy from
/// `to` - slice to copy to
///
/// # Example
///
/// ```ignore
/// copy_bulk(input.as_slice(), shared, 16);
/// ```
pub fn copy_bulk<C: CubePrimitive>(_from: &[C], _to: &mut [C], _length: usize) {}

pub mod copy_bulk {
    use cubecl_ir::{CopyMemoryOperands, Memory};

    use crate::ir::{Instruction, Scope};

    use super::*;

    /// The expand function for [`copy_bulk()`]
    pub fn expand<C: CubePrimitive>(
        scope: &Scope,
        from: &SliceExpand<C>,
        to: &mut SliceExpand<C>,
        length: usize,
    ) {
        let source = unsafe { *from.__expand_as_ptr_method(scope) }.expand;
        let target = unsafe { *to.__expand_as_mut_ptr_method(scope) }.expand;

        scope.register(Instruction::no_out(Memory::CopyMemory(
            CopyMemoryOperands {
                source,
                target,
                len: length,
            },
        )));
    }
}

/// Copy one element between two array-likes without intermediates.
///
/// # Arguments
///
/// `from` - The reference to copy from
/// `to` - The reference to copy to
///
/// # Example
///
/// ```ignore
/// copy(input[0], shared[0]);
/// ```
pub fn copy<C: CubePrimitive>(_from: &C, _to: &mut C) {}

pub mod copy {
    use cubecl_ir::{CopyMemoryOperands, Memory};

    use crate::ir::{Instruction, Scope};

    use super::*;

    /// The expand function for [`copy_bulk()`]
    pub fn expand<C: CubePrimitive>(
        scope: &Scope,
        from: &NativeExpand<C>,
        to: &mut NativeExpand<C>,
    ) {
        let source = from.expand;
        let target = to.expand;

        scope.register(Instruction::no_out(Memory::CopyMemory(
            CopyMemoryOperands {
                source,
                target,
                len: 1,
            },
        )));
    }
}