zyx 0.15.5

Zyx machine learning 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

// Copyright (C) 2025 zk4x
// SPDX-License-Identifier: LGPL-3.0-only

use crate::{
    Set,
    kernel::{Kernel, Op, OpId},
};

impl Kernel {
    pub fn instruction_schedule(&mut self) {
        let mut index_ops: Set<OpId> = Set::default();
        let mut insert_after: OpId = OpId::NULL;

        let mut op_id = self.head;
        while !op_id.is_null() {
            let next = self.next_op(op_id);
            match &self.ops[op_id].op {
                Op::Define { .. } => insert_after = op_id,
                Op::Const(_) | Op::Index { .. } | Op::Loop { .. } => {
                    index_ops.insert(op_id);
                    if !insert_after.is_null() && insert_after != op_id && !matches!(self.ops[op_id].op, Op::Loop { .. }) {
                        self.move_op_after(op_id, insert_after);
                    }
                    insert_after = op_id;
                }
                Op::Load { index, .. } => {
                    if index_ops.contains(index) && !insert_after.is_null() {
                        self.move_op_after(op_id, insert_after);
                        insert_after = op_id;
                    }
                }
                Op::Barrier { .. } | Op::Store { .. } | Op::EndLoop | Op::EndIf => {
                    insert_after = op_id;
                }
                _ => {
                    let params: Vec<OpId> = self.ops[op_id].op.parameters().collect();
                    if !params.is_empty() && params.iter().all(|p| index_ops.contains(p)) {
                        index_ops.insert(op_id);
                        if !insert_after.is_null() {
                            self.move_op_after(op_id, insert_after);
                        }
                        insert_after = op_id;
                    }
                }
            }
            op_id = next;
        }

        #[cfg(debug_assertions)]
        self.verify();
    }
}