qudit-core 0.3.2

Accelerated and Extensible Quantum Library
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
83

use std::ops::Mul;

// TODO: Add proper documentation to raw methods and add higher level
// functions that call them with the cartesian_match for loop unrolling.
/// Perform element-wise multiplication of two buffers.
///
/// # Safety
///
/// Caller must ensure that pointers point to properly addressable memory
/// that describe matrices with dimensions and strides given.
pub unsafe fn hadamard_kernel_raw<C: Mul<Output = C> + Copy>(
    nrows: usize,
    ncols: usize,
    dst: *mut C,
    dst_rs: isize,
    dst_cs: isize,
    lhs: *const C,
    lhs_rs: isize,
    lhs_cs: isize,
    rhs: *const C,
    rhs_rs: isize,
    rhs_cs: isize,
) {
    unsafe {
        let mut current_dst_ptr = dst;
        let mut current_lhs_ptr = lhs;
        let mut current_rhs_ptr = rhs;

        for _i in 0..nrows {
            current_dst_ptr = current_dst_ptr.offset(dst_rs);
            current_lhs_ptr = current_lhs_ptr.offset(lhs_rs);
            current_rhs_ptr = current_rhs_ptr.offset(rhs_rs);

            for _j in 0..ncols {
                *current_dst_ptr = *current_lhs_ptr * *current_rhs_ptr;

                current_dst_ptr = current_dst_ptr.offset(dst_cs);
                current_lhs_ptr = current_lhs_ptr.offset(lhs_cs);
                current_rhs_ptr = current_rhs_ptr.offset(rhs_cs);
            }
        }
    }
}

/// Perform element-wise multiplication of two buffers and add the result into output.
///
/// # Safety
///
/// Caller must ensure that pointers point to properly addressable memory
/// that describe matrices with dimensions and strides given.
pub unsafe fn hadamard_kernel_add_raw<C: Mul<Output = C> + Copy>(
    nrows: usize,
    ncols: usize,
    dst: *mut C,
    dst_rs: isize,
    dst_cs: isize,
    lhs: *const C,
    lhs_rs: isize,
    lhs_cs: isize,
    rhs: *const C,
    rhs_rs: isize,
    rhs_cs: isize,
) {
    unsafe {
        let mut current_dst_ptr = dst;
        let mut current_lhs_ptr = lhs;
        let mut current_rhs_ptr = rhs;

        for _i in 0..nrows {
            current_dst_ptr = current_dst_ptr.offset(dst_rs);
            current_lhs_ptr = current_lhs_ptr.offset(lhs_rs);
            current_rhs_ptr = current_rhs_ptr.offset(rhs_rs);

            for _j in 0..ncols {
                *current_dst_ptr = *current_lhs_ptr * *current_rhs_ptr;

                current_dst_ptr = current_dst_ptr.offset(dst_cs);
                current_lhs_ptr = current_lhs_ptr.offset(lhs_cs);
                current_rhs_ptr = current_rhs_ptr.offset(rhs_cs);
            }
        }
    }
}