mady_macro_core 0.1.0-alpha

🌲 mady's subpackage
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

use crate::graph::Node;
use crate::parser::{ParseGraph, VarType};
use proc_macro2::Ident;
use quote::format_ident;
use std::collections::hash_map::DefaultHasher;
use std::hash::{Hash, Hasher};

impl ParseGraph {
    pub fn grad_node(&self) -> Vec<Node> {
        self.nodes()
            .into_iter()
            .filter(|x| {
                if let VarType::Grad(_) = self.node_weight(*x) {
                    true
                } else {
                    false
                }
            })
            .collect()
    }
}

pub fn grad_method<T>(method_name: T) -> Ident
where
    T: ToString,
{
    format_ident!("grad_{}", method_name.to_string())
}

pub fn ops_to_string(op: &syn::BinOp) -> Option<&'static str> {
    Some(match op {
        syn::BinOp::Add(_) => "add",
        syn::BinOp::Sub(_) => "sub",
        syn::BinOp::Mul(_) => "mul",
        syn::BinOp::Div(_) => "div",
        // syn::BinOp::AddEq(_) => todo!(),
        // syn::BinOp::SubEq(_) => todo!(),
        // syn::BinOp::MulEq(_) => todo!(),
        // syn::BinOp::DivEq(_) => todo!(),
        _ => return None,
    })
}

pub fn into_hash<T>(v: &T) -> u64
where
    T: Hash,
{
    let mut hasher = DefaultHasher::new();
    v.hash(&mut hasher);
    hasher.finish()
}