# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * from .environments import * register_library(__name__, "pblas")
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library import barlib register_library(__name__, "FooLib")
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * register_library(__name__, "linalg")
from dace.library import register_library register_library(__name__, "BarLib")
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * from .environments import * from .utils import * register_library(__name__, "nccl")
# Copyright 2019-2020 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * from .environments import * register_library(__name__, "standard")
from dace.library import register_library from .environments import ONNXRuntime from .nodes import * from .schema import onnx_representation, ONNXAttributeType, ONNXAttribute, ONNXTypeConstraint, ONNXParameterType, ONNXSchema, ONNXParameter from .check_impl import check_op from .onnx_importer import ONNXModel register_library(__name__, "onnx")
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * from .environments import * from .utils import * register_library(__name__, "mpi")
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. import copy import dace from dace import SDFGState, SDFG, library, dtypes from dace.transformation.transformation import ExpandTransformation library.register_library(__name__, "AddLib") @dace.library.expansion class ExpandAdd(ExpandTransformation): environments = [] @staticmethod def expansion(node, parent_state: SDFGState, parent_sdfg: SDFG): in_edge = parent_state.in_edges(node)[0] out_edge = parent_state.out_edges(node)[0] sdfg = dace.SDFG("nested") sdfg.add_datadesc("_a", copy.deepcopy(parent_sdfg.arrays[in_edge.data.data])) sdfg.add_datadesc( "_b", copy.deepcopy(parent_sdfg.arrays[out_edge.data.data])) sdfg.arrays["_a"].transient = False sdfg.arrays["_b"].transient = False state = sdfg.add_state() inp = state.add_access("_a") outp = state.add_access("_b") me, mx = state.add_map("useless_map", {"i": "0"})
# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved. from dace.library import register_library from .nodes import * from .environments import * register_library(__name__, "lapack")