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
use super::super::*;
// ========================================================================
// Phase 3: Packing Tests
// ========================================================================
#[test]
fn test_pack_a_layout() {
// 4x3 matrix, pack first 4 rows
let a = vec![
1.0, 2.0, 3.0, // row 0
4.0, 5.0, 6.0, // row 1
7.0, 8.0, 9.0, // row 2
10.0, 11.0, 12.0, // row 3
];
let mut packed = vec![0.0; packed_a_size(4, 3)];
pack_a(&a, 3, 4, 3, &mut packed);
// Expected layout: column-major within MR-panels
// For MR=8, we have one panel with 4 real rows + 4 zero padding
// Col 0: [1, 4, 7, 10, 0, 0, 0, 0]
// Col 1: [2, 5, 8, 11, 0, 0, 0, 0]
// Col 2: [3, 6, 9, 12, 0, 0, 0, 0]
assert_eq!(packed[0], 1.0); // (0,0)
assert_eq!(packed[1], 4.0); // (1,0)
assert_eq!(packed[2], 7.0); // (2,0)
assert_eq!(packed[3], 10.0); // (3,0)
assert_eq!(packed[4], 0.0); // padding
assert_eq!(packed[MR], 2.0); // (0,1)
}
#[test]
fn test_pack_b_layout() {
// 3x4 matrix
let b = vec![
1.0, 2.0, 3.0, 4.0, // row 0
5.0, 6.0, 7.0, 8.0, // row 1
9.0, 10.0, 11.0, 12.0, // row 2
];
let mut packed = vec![0.0; packed_b_size(3, 4)];
pack_b(&b, 4, 3, 4, &mut packed);
// Expected: row-major within NR-panels
// For NR=6, we have one panel with 4 real cols + 2 zero padding
// Row 0: [1, 2, 3, 4, 0, 0]
// Row 1: [5, 6, 7, 8, 0, 0]
// Row 2: [9, 10, 11, 12, 0, 0]
assert_eq!(packed[0], 1.0);
assert_eq!(packed[1], 2.0);
assert_eq!(packed[2], 3.0);
assert_eq!(packed[3], 4.0);
assert_eq!(packed[4], 0.0); // padding
assert_eq!(packed[NR], 5.0); // row 1
}