cubecl-core 0.11.0-pre.4

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

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::dialect::memory::CopyOp;

    use crate::ir::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) }.value(scope);
        let target = unsafe { *to.__expand_as_mut_ptr_method(scope) }.value(scope);

        scope.register(&CopyOp::new(scope.ctx_mut(), source, target, 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::dialect::memory::CopyOp;

    use crate::ir::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.value(scope);
        let target = to.value(scope);

        scope.register(&CopyOp::new(scope.ctx_mut(), source, target, 1));
    }
}