ntops/src/ntops/kernels/__init__.py at 1ce855a6f13eb6a536aef36508cb8c089b0dbf7c · 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
90
91
92
93
94
95
from ntops.kernels import (
    abs,
    add,
    addmm,
    alpha_dropout,
    avg_pool2d,
    bitwise_and,
    bitwise_not,
    bitwise_or,
    bmm,
    clamp,
    conv2d,
    cos,
    cosh,
    diag,
    div,
    dropout,
    eq,
    exp,
    ge,
    gelu,
    gt,
    isinf,
    isnan,
    layer_norm,
    le,
    lt,
    max_pool2d,
    mm,
    mul,
    ne,
    neg,
    pow,
    relu,
    rms_norm,
    rotary_position_embedding,
    round,
    rsqrt,
    scaled_dot_product_attention,
    sigmoid,
    silu,
    sin,
    softmax,
    sort,
    sub,
    tanh,
)

__all__ = [
    "abs",
    "add",
    "addmm",
    "alpha_dropout",
    "avg_pool2d",
    "bitwise_and",
    "bitwise_not",
    "bitwise_or",
    "bmm",
    "clamp",
    "conv2d",
    "cos",
    "cosh",
    "diag",
    "div",
    "dropout",
    "eq",
    "exp",
    "ge",
    "gelu",
    "gt",
    "isinf",
    "isnan",
    "layer_norm",
    "le",
    "lt",
    "max_pool2d",
    "mm",
    "mul",
    "ne",
    "neg",
    "pow",
    "relu",
    "rms_norm",
    "rotary_position_embedding",
    "round",
    "rsqrt",
    "scaled_dot_product_attention",
    "sigmoid",
    "silu",
    "sin",
    "softmax",
    "sort",
    "sub",
    "tanh",
]