dace.frontend package

Subpackages

• dace.frontend.common package
  • Submodules
  • dace.frontend.common.op_repository module
    • Replacements
      • Replacements.get()
      • Replacements.get_attribute()
      • Replacements.get_method()
      • Replacements.get_ufunc()
      • Replacements.getop()
    • replaces()
    • replaces_attribute()
    • replaces_method()
    • replaces_operator()
    • replaces_ufunc()
  • Module contents
• dace.frontend.octave package
  • Submodules
  • dace.frontend.octave.ast_arrayaccess module
    • AST_ArrayAccess
      • AST_ArrayAccess.generate_code()
      • AST_ArrayAccess.get_basetype()
      • AST_ArrayAccess.get_children()
      • AST_ArrayAccess.get_dims()
      • AST_ArrayAccess.is_data_dependent_access()
      • AST_ArrayAccess.make_range_from_accdims()
      • AST_ArrayAccess.replace_child()
  • dace.frontend.octave.ast_assign module
    • AST_Assign
      • AST_Assign.defined_variables()
      • AST_Assign.generate_code()
      • AST_Assign.get_children()
      • AST_Assign.print_nodes()
      • AST_Assign.provide_parents()
      • AST_Assign.replace_child()
  • dace.frontend.octave.ast_expression module
    • AST_BinExpression
      • AST_BinExpression.generate_code()
      • AST_BinExpression.get_basetype()
      • AST_BinExpression.get_children()
      • AST_BinExpression.get_dims()
      • AST_BinExpression.matrix2d_matrix2d_mult()
      • AST_BinExpression.matrix2d_matrix2d_plus_or_minus()
      • AST_BinExpression.matrix2d_scalar()
      • AST_BinExpression.provide_parents()
      • AST_BinExpression.replace_child()
      • AST_BinExpression.scalar_scalar()
      • AST_BinExpression.vec_mult_vect()
    • AST_UnaryExpression
      • AST_UnaryExpression.get_children()
      • AST_UnaryExpression.replace_child()
      • AST_UnaryExpression.specialize()
  • dace.frontend.octave.ast_function module
    • AST_Argument
      • AST_Argument.get_children()
    • AST_BuiltInFunCall
      • AST_BuiltInFunCall.generate_code()
      • AST_BuiltInFunCall.get_basetype()
      • AST_BuiltInFunCall.get_children()
      • AST_BuiltInFunCall.get_dims()
      • AST_BuiltInFunCall.replace_child()
      • AST_BuiltInFunCall.specialize()
    • AST_EndFunc
      • AST_EndFunc.generate_code()
      • AST_EndFunc.get_children()
      • AST_EndFunc.replace_child()
    • AST_FunCall
      • AST_FunCall.get_children()
      • AST_FunCall.replace_child()
      • AST_FunCall.specialize()
    • AST_Function
      • AST_Function.generate_code()
      • AST_Function.get_children()
      • AST_Function.replace_child()
      • AST_Function.set_statements()
  • dace.frontend.octave.ast_loop module
    • AST_ForLoop
      • AST_ForLoop.generate_code()
      • AST_ForLoop.generate_code_proper()
      • AST_ForLoop.get_children()
      • AST_ForLoop.replace_child()
  • dace.frontend.octave.ast_matrix module
    • AST_Matrix
      • AST_Matrix.generate_code()
      • AST_Matrix.get_basetype()
      • AST_Matrix.get_children()
      • AST_Matrix.get_dims()
      • AST_Matrix.get_values_row_major()
      • AST_Matrix.is_constant()
      • AST_Matrix.provide_parents()
      • AST_Matrix.replace_child()
    • AST_Matrix_Row
      • AST_Matrix_Row.get_children()
      • AST_Matrix_Row.get_dims()
      • AST_Matrix_Row.is_constant()
      • AST_Matrix_Row.provide_parents()
      • AST_Matrix_Row.replace_child()
    • AST_Transpose
      • AST_Transpose.generate_code()
      • AST_Transpose.get_basetype()
      • AST_Transpose.get_children()
      • AST_Transpose.get_dims()
      • AST_Transpose.replace_child()
  • dace.frontend.octave.ast_node module
    • AST_Node
      • AST_Node.defined_variables()
      • AST_Node.find_data_node_in_sdfg_state()
      • AST_Node.generate_code()
      • AST_Node.get_children()
      • AST_Node.get_datanode()
      • AST_Node.get_initializers()
      • AST_Node.get_name_in_sdfg()
      • AST_Node.get_new_tmpvar()
      • AST_Node.get_parent()
      • AST_Node.print_as_tree()
      • AST_Node.provide_parents()
      • AST_Node.replace_child()
      • AST_Node.replace_parent()
      • AST_Node.search_vardef_in_scope()
      • AST_Node.shortdesc()
      • AST_Node.specialize()
    • AST_Statements
      • AST_Statements.append_statement()
      • AST_Statements.generate_code()
      • AST_Statements.get_children()
      • AST_Statements.provide_parents()
      • AST_Statements.replace_child()
      • AST_Statements.specialize()
  • dace.frontend.octave.ast_nullstmt module
    • AST_Comment
      • AST_Comment.generate_code()
      • AST_Comment.get_children()
      • AST_Comment.replace_child()
    • AST_EndStmt
      • AST_EndStmt.get_children()
      • AST_EndStmt.replace_child()
    • AST_NullStmt
      • AST_NullStmt.generate_code()
      • AST_NullStmt.get_children()
      • AST_NullStmt.replace_child()
  • dace.frontend.octave.ast_range module
    • AST_RangeExpression
      • AST_RangeExpression.generate_code()
      • AST_RangeExpression.get_basetype()
      • AST_RangeExpression.get_children()
      • AST_RangeExpression.get_dims()
      • AST_RangeExpression.replace_child()
      • AST_RangeExpression.specialize()
  • dace.frontend.octave.ast_values module
    • AST_Constant
      • AST_Constant.generate_code()
      • AST_Constant.get_basetype()
      • AST_Constant.get_children()
      • AST_Constant.get_dims()
      • AST_Constant.get_value()
      • AST_Constant.is_constant()
      • AST_Constant.replace_child()
    • AST_Ident
      • AST_Ident.generate_code()
      • AST_Ident.get_basetype()
      • AST_Ident.get_children()
      • AST_Ident.get_dims()
      • AST_Ident.get_name()
      • AST_Ident.get_name_in_sdfg()
      • AST_Ident.get_propagated_value()
      • AST_Ident.is_constant()
      • AST_Ident.replace_child()
      • AST_Ident.specialize()
  • dace.frontend.octave.lexer module
    • main()
    • new()
    • raise_exception()
  • dace.frontend.octave.parse module
    • p_arg1()
    • p_arg2()
    • p_arg_list()
    • p_args()
    • p_break_stmt()
    • p_case_list()
    • p_cellarray()
    • p_cellarray_2()
    • p_cellarrayref()
    • p_command()
    • p_comment_stmt()
    • p_concat_list1()
    • p_concat_list2()
    • p_continue_stmt()
    • p_elseif_stmt()
    • p_end()
    • p_end_function()
    • p_error()
    • p_error_stmt()
    • p_expr()
    • p_expr1()
    • p_expr2()
    • p_expr_2()
    • p_expr_colon()
    • p_expr_end()
    • p_expr_ident()
    • p_expr_list()
    • p_expr_number()
    • p_expr_stmt()
    • p_expr_string()
    • p_exprs()
    • p_field_expr()
    • p_foo_stmt()
    • p_for_stmt()
    • p_func_stmt()
    • p_funcall_expr()
    • p_global()
    • p_global_list()
    • p_global_stmt()
    • p_ident_init_opt()
    • p_if_stmt()
    • p_lambda_args()
    • p_lambda_expr()
    • p_matrix()
    • p_matrix_2()
    • p_null_stmt()
    • p_parens_expr()
    • p_persistent_stmt()
    • p_ret()
    • p_return_stmt()
    • p_semi_opt()
    • p_separator()
    • p_stmt()
    • p_stmt_list()
    • p_stmt_list_opt()
    • p_switch_stmt()
    • p_top()
    • p_transpose_expr()
    • p_try_catch()
    • p_unwind()
    • p_while_stmt()
    • parse()
  • dace.frontend.octave.parsetab module
  • Module contents
• dace.frontend.python package
  • Submodules
  • dace.frontend.python.astutils module
    • ASTFindReplace
      • ASTFindReplace.visit_Name()
      • ASTFindReplace.visit_keyword()
    • ASTHelperMixin
      • ASTHelperMixin.generic_visit_field()
      • ASTHelperMixin.generic_visit_filtered()
    • AnnotateTopLevel
      • AnnotateTopLevel.visit_TopLevel()
    • ConstantExtractor
      • ConstantExtractor.visit_Constant()
      • ConstantExtractor.visit_Name()
      • ConstantExtractor.visit_Num()
    • ExtNodeTransformer
      • ExtNodeTransformer.generic_visit()
      • ExtNodeTransformer.visit_TopLevel()
    • ExtNodeVisitor
      • ExtNodeVisitor.generic_visit()
      • ExtNodeVisitor.visit_TopLevel()
    • ExtUnparser
    • NameFound
    • RemoveSubscripts
      • RemoveSubscripts.visit_Subscript()
    • TaskletFreeSymbolVisitor
      • TaskletFreeSymbolVisitor.visit_AnnAssign()
      • TaskletFreeSymbolVisitor.visit_Attribute()
      • TaskletFreeSymbolVisitor.visit_Call()
      • TaskletFreeSymbolVisitor.visit_Name()
    • astrange_to_symrange()
    • copy_tree()
    • create_constant()
    • escape_string()
    • evalnode()
    • function_to_ast()
    • is_constant()
    • negate_expr()
    • parse_function_arguments()
    • rname()
    • slice_to_subscript()
    • subscript_to_ast_slice()
    • subscript_to_ast_slice_recursive()
    • subscript_to_slice()
    • unparse()
  • dace.frontend.python.cached_program module
    • DaceProgramCache
      • DaceProgramCache.add()
      • DaceProgramCache.clear()
      • DaceProgramCache.get()
      • DaceProgramCache.has()
      • DaceProgramCache.make_key()
      • DaceProgramCache.pop()
    • LimitedSizeDict
    • ProgramCacheEntry
      • ProgramCacheEntry.compiled_sdfg
      • ProgramCacheEntry.sdfg
    • ProgramCacheKey
      • ProgramCacheKey.arg_types
      • ProgramCacheKey.closure_constants
      • ProgramCacheKey.closure_types
      • ProgramCacheKey.specified_args
  • dace.frontend.python.common module
    • DaceSyntaxError
    • SDFGClosure
      • SDFGClosure.array_mapping
      • SDFGClosure.call_tree_length()
      • SDFGClosure.callbacks
      • SDFGClosure.callstack
      • SDFGClosure.closure_arrays
      • SDFGClosure.closure_constants
      • SDFGClosure.closure_sdfgs
      • SDFGClosure.combine_nested_closures()
      • SDFGClosure.nested_closures
      • SDFGClosure.print_call_tree()
    • SDFGConvertible
      • SDFGConvertible.closure_resolver()
    • StringLiteral
      • StringLiteral.value
    • inverse_dict_lookup()
  • dace.frontend.python.interface module
    • MapGenerator
    • MapMetaclass
    • TaskletMetaclass
    • consume
    • function()
    • in_program()
    • inline()
    • map
    • method()
    • nounroll()
    • program()
    • tasklet
    • unroll()
  • dace.frontend.python.memlet_parser module
    • MemletExpr
      • MemletExpr.accesses
      • MemletExpr.arrdims
      • MemletExpr.name
      • MemletExpr.new_axes
      • MemletExpr.subset
      • MemletExpr.wcr
    • ParseMemlet()
    • inner_eval_ast()
    • parse_memlet()
    • parse_memlet_subset()
    • pyexpr_to_symbolic()
  • dace.frontend.python.ndloop module
    • NDLoop()
    • ndrange()
    • slicetoxrange()
  • dace.frontend.python.newast module
    • AddTransientMethods
      • AddTransientMethods.get()
    • ProgramVisitor
      • ProgramVisitor.create_callback()
      • ProgramVisitor.defined
      • ProgramVisitor.increment_progress()
      • ProgramVisitor.make_slice()
      • ProgramVisitor.parse_program()
      • ProgramVisitor.progress_bar
      • ProgramVisitor.progress_count()
      • ProgramVisitor.start_time
      • ProgramVisitor.visit()
      • ProgramVisitor.visit_AnnAssign()
      • ProgramVisitor.visit_Assign()
      • ProgramVisitor.visit_AsyncWith()
      • ProgramVisitor.visit_Attribute()
      • ProgramVisitor.visit_AugAssign()
      • ProgramVisitor.visit_BinOp()
      • ProgramVisitor.visit_BoolOp()
      • ProgramVisitor.visit_Break()
      • ProgramVisitor.visit_Bytes()
      • ProgramVisitor.visit_Call()
      • ProgramVisitor.visit_Compare()
      • ProgramVisitor.visit_Constant()
      • ProgramVisitor.visit_Continue()
      • ProgramVisitor.visit_Dict()
      • ProgramVisitor.visit_ExtSlice()
      • ProgramVisitor.visit_For()
      • ProgramVisitor.visit_FunctionDef()
      • ProgramVisitor.visit_If()
      • ProgramVisitor.visit_Index()
      • ProgramVisitor.visit_Lambda()
      • ProgramVisitor.visit_List()
      • ProgramVisitor.visit_Name()
      • ProgramVisitor.visit_NameConstant()
      • ProgramVisitor.visit_NamedExpr()
      • ProgramVisitor.visit_Num()
      • ProgramVisitor.visit_Return()
      • ProgramVisitor.visit_Set()
      • ProgramVisitor.visit_Str()
      • ProgramVisitor.visit_Subscript()
      • ProgramVisitor.visit_TopLevelExpr()
      • ProgramVisitor.visit_Tuple()
      • ProgramVisitor.visit_TypeAlias()
      • ProgramVisitor.visit_UnaryOp()
      • ProgramVisitor.visit_While()
      • ProgramVisitor.visit_With()
    • SkipCall
    • TaskletTransformer
      • TaskletTransformer.parse_tasklet()
      • TaskletTransformer.visit_Call()
      • TaskletTransformer.visit_Name()
      • TaskletTransformer.visit_TopLevelExpr()
      • TaskletTransformer.visit_TopLevelStr()
    • add_indirection_subgraph()
    • parse_dace_program()
    • specifies_datatype()
    • until()
  • dace.frontend.python.parser module
    • DaceProgram
      • DaceProgram.auto_optimize()
      • DaceProgram.closure_resolver()
      • DaceProgram.compile()
      • DaceProgram.get_program_hash()
      • DaceProgram.load_precompiled_sdfg()
      • DaceProgram.load_sdfg()
      • DaceProgram.methodobj
      • DaceProgram.name
      • DaceProgram.to_sdfg()
      • DaceProgram.validate
    • infer_symbols_from_datadescriptor()
  • dace.frontend.python.preprocessing module
    • ArrayClosureResolver
      • ArrayClosureResolver.visit_Name()
    • AugAssignExpander
      • AugAssignExpander.visit_AugAssign()
    • CallTreeResolver
      • CallTreeResolver.visit_Call()
    • ConditionalCodeResolver
      • ConditionalCodeResolver.visit_If()
      • ConditionalCodeResolver.visit_IfExp()
      • ConditionalCodeResolver.visit_Name()
    • ContextManagerInliner
      • ContextManagerInliner.visit_AsyncFor()
      • ContextManagerInliner.visit_AsyncWith()
      • ContextManagerInliner.visit_Break()
      • ContextManagerInliner.visit_Continue()
      • ContextManagerInliner.visit_For()
      • ContextManagerInliner.visit_Return()
      • ContextManagerInliner.visit_While()
      • ContextManagerInliner.visit_With()
    • DaceRecursionError
    • DeadCodeEliminator
      • DeadCodeEliminator.generic_visit()
    • DisallowedAssignmentChecker
      • DisallowedAssignmentChecker.visit_AnnAssign()
      • DisallowedAssignmentChecker.visit_Assign()
      • DisallowedAssignmentChecker.visit_AugAssign()
      • DisallowedAssignmentChecker.visit_Call()
      • DisallowedAssignmentChecker.visit_NamedExpr()
    • ExpressionInliner
      • ExpressionInliner.visit_Call()
    • GlobalResolver
      • GlobalResolver.generic_visit()
      • GlobalResolver.global_value_to_node()
      • GlobalResolver.globals
      • GlobalResolver.visit_Assert()
      • GlobalResolver.visit_AsyncFunctionDef()
      • GlobalResolver.visit_Attribute()
      • GlobalResolver.visit_AugAssign()
      • GlobalResolver.visit_Call()
      • GlobalResolver.visit_For()
      • GlobalResolver.visit_FunctionDef()
      • GlobalResolver.visit_JoinedStr()
      • GlobalResolver.visit_Lambda()
      • GlobalResolver.visit_Name()
      • GlobalResolver.visit_Raise()
      • GlobalResolver.visit_Subscript()
      • GlobalResolver.visit_TopLevelExpr()
      • GlobalResolver.visit_keyword()
    • LoopUnroller
      • LoopUnroller.STATELESS_GENERATORS
      • LoopUnroller.THRESHOLD_GENERATORS
      • LoopUnroller.visit_AsyncFor()
      • LoopUnroller.visit_For()
    • MPIResolver
      • MPIResolver.visit()
      • MPIResolver.visit_Attribute()
      • MPIResolver.visit_Name()
    • ModuleResolver
      • ModuleResolver.visit_Attribute()
      • ModuleResolver.visit_Call()
    • ModuloConverter
      • ModuloConverter.visit_BinOp()
    • PreprocessedAST
      • PreprocessedAST.filename
      • PreprocessedAST.preprocessed_ast
      • PreprocessedAST.program_globals
      • PreprocessedAST.src
      • PreprocessedAST.src_line
    • RewriteSympyEquality
      • RewriteSympyEquality.visit_Compare()
      • RewriteSympyEquality.visit_Constant()
      • RewriteSympyEquality.visit_Num()
    • StructTransformer
      • StructTransformer.visit_Call()
    • find_disallowed_statements()
    • flatten_callback()
    • has_replacement()
    • preprocess_dace_program()
  • dace.frontend.python.replacements module
    • dot()
    • eye()
    • flat()
    • implement_ufunc()
    • implement_ufunc_accumulate()
    • implement_ufunc_outer()
    • implement_ufunc_reduce()
    • normalize_axes()
    • reshape()
    • size()
    • view()
  • dace.frontend.python.tasklet_runner module
    • TaskletRewriter
      • TaskletRewriter.clear_statements()
      • TaskletRewriter.rewrite_tasklet()
      • TaskletRewriter.visit_Assign()
      • TaskletRewriter.visit_Name()
      • TaskletRewriter.visit_TopLevelExpr()
    • get_tasklet_ast()
    • run_tasklet()
  • dace.frontend.python.wrappers module
    • define_local()
    • define_local_scalar()
    • define_stream()
    • define_streamarray()
    • ndarray()
    • scalar()
    • stream_array
      • stream_array.shape
  • Module contents

Submodules

dace.frontend.operations module

class dace.frontend.operations.CompiledSDFGProfiler(repetitions=0, warmup=0, tqdm_leave=True, print_results=True)

Bases: object

A context manager that prints the time it takes to execute the generated SDFG code (excluding init and shutdown).

times: List[Tuple[SDFG, List[float]]]

The list of SDFGs and times for each SDFG called within the context.

dace.frontend.operations.detect_reduction_type(wcr_str, openmp=False)

Inspects a lambda function and tries to determine if it’s one of the built-in reductions that frameworks such as MPI can provide.

Parameters:

• wcr_str – A Python string representation of the lambda function.

• openmp – Detect additional OpenMP reduction types.

Returns:

dtypes.ReductionType if detected, dtypes.ReductionType.Custom if not detected, or None if no reduction is found.

dace.frontend.operations.elementwise(func, in_array, out_array=None)

Applies a function to each element of the array

Parameters:

• in_array – array to apply to.

• out_array – output array to write the result to. If None, a new array will be returned

• func – lambda function to apply to each element.

Returns:

new array with the lambda applied to each element

dace.frontend.operations.is_op_associative(wcr_str)

Inspects a custom lambda function and tries to determine whether it is symbolically associative (disregarding data type).

Parameters:

wcr_str – A string in Python representing a lambda function.

Returns:

True if associative, False if not, None if cannot be determined.

dace.frontend.operations.is_op_commutative(wcr_str)

Inspects a custom lambda function and tries to determine whether it is symbolically commutative (disregarding data type).

Parameters:

wcr_str – A string in Python representing a lambda function.

Returns:

True if commutative, False if not, None if cannot be determined.

dace.frontend.operations.reduce(op, in_array, out_array=None, axis=None, identity=None)

Reduces an array according to a binary operation op, starting with initial value identity, over the given axis (or axes if axis is a list), to out_array.

Requires out_array with len(axis) dimensions less than in_array, or a scalar if axis is None.

Parameters:

• op – binary operation to use for reduction.

• in_array – array to reduce.

• out_array – output array to write the result to. If None, a new array will be returned.

• axis – the axis or axes to reduce over. If None, all axes will be reduced.

• identity – intial value for the reduction. If None, uses value stored in output.

Returns:

None if out_array is given, or the newly created out_array if out_array is None.

Module contents