def visit_BlockStmt(self, node: nir.BlockStmt, **kwargs): statements = [] for decl in node.declarations: statements.append( usid.VarDecl( name=decl.name, init=usid.Literal(value="0.0", vtype=decl.vtype, location_type=node.location_type), vtype=decl.vtype, location_type=node.location_type, )) for stmt in node.statements: statements.append(self.visit(stmt, **kwargs)) return statements
def visit_Literal(self, node: nir.Literal, **kwargs): return usid.Literal(value=node.value, vtype=node.vtype, location_type=node.location_type)