Esempi in Python per register_library

Esempio n. 1

File: __init__.py Progetto: zurvar/dace

# 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")

Esempio n. 2

File: __init__.py Progetto: zurvar/dace

# Copyright 2019-2021 ETH Zurich and the DaCe authors. All rights reserved.
from dace.library import register_library
import barlib

register_library(__name__, "FooLib")

Esempio n. 3

File: __init__.py Progetto: zurvar/dace

# 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")

Esempio n. 4

from dace.library import register_library

register_library(__name__, "BarLib")

Esempio n. 5

# 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")

Esempio n. 6

# 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")

Esempio n. 7

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")

Esempio n. 8

File: __init__.py Progetto: zurvar/dace

# 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")

Esempio n. 9

# 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"})

Esempio n. 10

# 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")