pub fn extract_device_functions(ast: &Ast) -> Vec<&FunctionDef>
Extract all device functions from the AST