bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Single commit implementing the enzyme/autodiff frontend

Browse source 
Co-authored-by: Lorenz Schmidt <bytesnake@mailbox.org>
This commit is contained in:
Manuel Drehwald 2024-10-11 19:13:31 +02:00
parent 52fd998399
commit 624c071b99
17 changed files with 1384 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
compiler/rustc_expand/src/build.rs
View file

@ -220,6 +220,10 @@ impl<'a> ExtCtxt<'a> {
self.stmt_local(local, span)
}
pub fn stmt_semi(&self, expr: P<ast::Expr>) -> ast::Stmt {
ast::Stmt { id: ast::DUMMY_NODE_ID, span: expr.span, kind: ast::StmtKind::Semi(expr) }
}
pub fn stmt_local(&self, local: P<ast::Local>, span: Span) -> ast::Stmt {
ast::Stmt { id: ast::DUMMY_NODE_ID, kind: ast::StmtKind::Let(local), span }
}
@ -287,6 +291,25 @@ impl<'a> ExtCtxt<'a> {
self.expr(sp, ast::ExprKind::Paren(e))
}
pub fn expr_method_call(
&self,
span: Span,
expr: P<ast::Expr>,
ident: Ident,
args: ThinVec<P<ast::Expr>>,
) -> P<ast::Expr> {
let seg = ast::PathSegment::from_ident(ident);
self.expr(
span,
ast::ExprKind::MethodCall(Box::new(ast::MethodCall {
seg,
receiver: expr,
args,
span,
})),
)
}
pub fn expr_call(
&self,
span: Span,
@ -295,6 +318,12 @@ impl<'a> ExtCtxt<'a> {
) -> P<ast::Expr> {
self.expr(span, ast::ExprKind::Call(expr, args))
}
pub fn expr_loop(&self, sp: Span, block: P<ast::Block>) -> P<ast::Expr> {
self.expr(sp, ast::ExprKind::Loop(block, None, sp))
}
pub fn expr_asm(&self, sp: Span, expr: P<ast::InlineAsm>) -> P<ast::Expr> {
self.expr(sp, ast::ExprKind::InlineAsm(expr))
}
pub fn expr_call_ident(
&self,
span: Span,