InfiniCore/python/infinicore/__init__.py at 5f3bc5796fddf5f22a30b7b8748fd72db5e5d9a9 · InfiniTensor/InfiniCore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from infinicore.device import device
from infinicore.dtype import (
    bfloat16,
    bool,
    cdouble,
    cfloat,
    chalf,
    complex32,
    complex64,
    complex128,
    double,
    dtype,
    float,
    float16,
    float32,
    float64,
    half,
    int,
    int8,
    int16,
    int32,
    int64,
    long,
    short,
    uint8,
)
from infinicore.ntops import use_ntops
from infinicore.ops.matmul import matmul
from infinicore.ops.rearrange import rearrange
from infinicore.tensor import (
    empty,
    from_blob,
    ones,
    strided_empty,
    strided_from_blob,
    zeros,
)

__all__ = [
    # Classes.
    "device",
    "dtype",
    # Data Types.
    "bfloat16",
    "bool",
    "cdouble",
    "cfloat",
    "chalf",
    "complex32",
    "complex64",
    "complex128",
    "double",
    "float",
    "float16",
    "float32",
    "float64",
    "half",
    "int",
    "int8",
    "int16",
    "int32",
    "int64",
    "long",
    "short",
    "uint8",
    # `ntops` integration.
    "use_ntops",
    # Operations.
    "matmul",
    "rearrange",
    "empty",
    "from_blob",
    "ones",
    "strided_empty",
    "strided_from_blob",
    "zeros",
]