ntops/src/ntops/kernels/__init__.py at c5d68d2e832f2ac664e60abee2fb01a3c63a683e · InfiniTensor/ntops · 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
78
79
80
81
82
83
84
85
86
87
88
89
from ntops.kernels import (
    abs,
    add,
    addmm,
    bitwise_and,
    bitwise_not,
    bitwise_or,
    bmm,
    clamp,
    cos,
    div,
    dropout,
    eq,
    exp,
    ge,
    gelu,
    gt,
    isinf,
    isnan,
    layer_norm,
    le,
    lt,
    mm,
    mul,
    ne,
    neg,
    pow,
    relu,
    rms_norm,
    rotary_position_embedding,
    rsqrt,
    scaled_dot_product_attention,
    sigmoid,
    silu,
    sin,
    softmax,
    sub,
    tanh,
    bitwise_left_shift,
    index_select,
    fold,
    mish,
    log2,
)

__all__ = [
    "abs",
    "add",
    "addmm",
    "bitwise_and",
    "bitwise_not",
    "bitwise_or",
    "bmm",
    "clamp",
    "cos",
    "div",
    "dropout",
    "eq",
    "exp",
    "ge",
    "gelu",
    "gt",
    "isinf",
    "isnan",
    "layer_norm",
    "le",
    "lt",
    "mm",
    "mul",
    "ne",
    "neg",
    "pow",
    "relu",
    "rms_norm",
    "rotary_position_embedding",
    "rsqrt",
    "scaled_dot_product_attention",
    "sigmoid",
    "silu",
    "sin",
    "softmax",
    "sub",
    "tanh",
    "bitwise_left_shift",
    "index_select",
    "fold",
    "mish",
    "log2",
]