-
Notifications
You must be signed in to change notification settings - Fork 110
Expand file tree
/
Copy pathinfiniop.h
More file actions
108 lines (107 loc) · 3.8 KB
/
infiniop.h
File metadata and controls
108 lines (107 loc) · 3.8 KB
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
96
97
98
99
100
101
102
103
104
105
106
107
108
#ifndef __INFINIOP_API_H__
#define __INFINIOP_API_H__
#include "infiniop/handle.h"
#include "infiniop/ops/acos.h"
#include "infiniop/ops/adaptive_avg_pool1d.h"
#include "infiniop/ops/adaptive_avg_pool3d.h"
#include "infiniop/ops/adaptive_max_pool1d.h"
#include "infiniop/ops/add.h"
#include "infiniop/ops/add_rms_norm.h"
#include "infiniop/ops/addbmm.h"
#include "infiniop/ops/addcmul.h"
#include "infiniop/ops/addr.h"
#include "infiniop/ops/affine_grid.h"
#include "infiniop/ops/all.h"
#include "infiniop/ops/asin.h"
#include "infiniop/ops/asinh.h"
#include "infiniop/ops/atanh.h"
#include "infiniop/ops/attention.h"
#include "infiniop/ops/avg_pool1d.h"
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
#include "infiniop/ops/broadcast_to.h"
#include "infiniop/ops/causal_softmax.h"
#include "infiniop/ops/cdist.h"
#include "infiniop/ops/clip.h"
#include "infiniop/ops/conv.h"
#include "infiniop/ops/cross_entropy.h"
#include "infiniop/ops/dequant/per_tensor_dequant_int8.h"
#include "infiniop/ops/dequantize_awq.h"
#include "infiniop/ops/dequantize_gptq.h"
#include "infiniop/ops/embedding.h"
#include "infiniop/ops/equal.h"
#include "infiniop/ops/erf.h"
#include "infiniop/ops/erfc.h"
#include "infiniop/ops/erfinv.h"
#include "infiniop/ops/flash_attention.h"
#include "infiniop/ops/flipud.h"
#include "infiniop/ops/float_power.h"
#include "infiniop/ops/floor.h"
#include "infiniop/ops/floor_divide.h"
#include "infiniop/ops/fmin.h"
#include "infiniop/ops/fmod.h"
#include "infiniop/ops/gelu.h"
#include "infiniop/ops/gemm.h"
#include "infiniop/ops/hardswish.h"
#include "infiniop/ops/hardtanh.h"
#include "infiniop/ops/huber_loss.h"
#include "infiniop/ops/hypot.h"
#include "infiniop/ops/index_add.h"
#include "infiniop/ops/index_copy.h"
#include "infiniop/ops/inner.h"
#include "infiniop/ops/int8_gemm.h"
#include "infiniop/ops/kthvalue.h"
#include "infiniop/ops/kv_caching.h"
#include "infiniop/ops/layer_norm.h"
#include "infiniop/ops/ldexp.h"
#include "infiniop/ops/lerp.h"
#include "infiniop/ops/log_softmax.h"
#include "infiniop/ops/logaddexp.h"
#include "infiniop/ops/logaddexp2.h"
#include "infiniop/ops/logcumsumexp.h"
#include "infiniop/ops/lp_norm.h"
#include "infiniop/ops/masked_select.h"
#include "infiniop/ops/matrix_power.h"
#include "infiniop/ops/mul.h"
#include "infiniop/ops/multi_margin_loss.h"
#include "infiniop/ops/ones.h"
#include "infiniop/ops/paged_attention.h"
#include "infiniop/ops/paged_attention_prefill.h"
#include "infiniop/ops/paged_caching.h"
#include "infiniop/ops/pixel_shuffle.h"
#include "infiniop/ops/quant/per_channel_quant_int8.h"
#include "infiniop/ops/quant/per_tensor_quant_int8.h"
#include "infiniop/ops/random_sample.h"
#include "infiniop/ops/rearrange.h"
#include "infiniop/ops/reciprocal.h"
#include "infiniop/ops/relu.h"
#include "infiniop/ops/rms_norm.h"
#include "infiniop/ops/rope.h"
#include "infiniop/ops/scatter.h"
#include "infiniop/ops/sigmoid.h"
#include "infiniop/ops/silu.h"
#include "infiniop/ops/silu_and_mul.h"
#include "infiniop/ops/smooth_l1_loss.h"
#include "infiniop/ops/softmax.h"
#include "infiniop/ops/softplus.h"
#include "infiniop/ops/softsign.h"
#include "infiniop/ops/sub.h"
#include "infiniop/ops/sum.h"
#include "infiniop/ops/swiglu.h"
#include "infiniop/ops/take.h"
#include "infiniop/ops/tan.h"
#include "infiniop/ops/tanh.h"
#include "infiniop/ops/tanhshrink.h"
#include "infiniop/ops/topk.h"
#include "infiniop/ops/topkrouter.h"
#include "infiniop/ops/topksoftmax.h"
#include "infiniop/ops/triplet_margin_loss.h"
#include "infiniop/ops/triplet_margin_with_distance_loss.h"
#include "infiniop/ops/unfold.h"
#include "infiniop/ops/upsample_bilinear.h"
#include "infiniop/ops/upsample_nearest.h"
#include "infiniop/ops/vander.h"
#include "infiniop/ops/var.h"
#include "infiniop/ops/var_mean.h"
#include "infiniop/ops/zeros.h"
#include "infiniop/tensor_descriptor.h"
#endif // __INFINIOP_API_H__