infiniop.h 2 KB
Newer Older
PanZezhong's avatar
PanZezhong committed
1
2
#ifndef __INFINIOP_API_H__
#define __INFINIOP_API_H__
3

4
#include "infiniop/handle.h"
PanZezhongQY's avatar
PanZezhongQY committed
5
#include "infiniop/ops/add.h"
6
#include "infiniop/ops/add_rms_norm.h"
PanZezhong's avatar
PanZezhong committed
7
#include "infiniop/ops/addcmul.h"
PanZezhong's avatar
PanZezhong committed
8
#include "infiniop/ops/all.h"
PanZezhong's avatar
PanZezhong committed
9
#include "infiniop/ops/atanh.h"
PanZezhongQY's avatar
PanZezhongQY committed
10
#include "infiniop/ops/attention.h"
PanZezhong's avatar
PanZezhong committed
11
12
#include "infiniop/ops/avg_pool1d.h"
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
PanZezhongQY's avatar
PanZezhongQY committed
13
#include "infiniop/ops/causal_softmax.h"
PanZezhong's avatar
PanZezhong committed
14
#include "infiniop/ops/cdist.h"
goldenfox2025's avatar
goldenfox2025 committed
15
#include "infiniop/ops/clip.h"
PanZezhongQY's avatar
PanZezhongQY committed
16
#include "infiniop/ops/conv.h"
PanZezhong's avatar
PanZezhong committed
17
#include "infiniop/ops/cross_entropy.h"
18
#include "infiniop/ops/dequantize_awq.h"
19
#include "infiniop/ops/embedding.h"
PanZezhong's avatar
PanZezhong committed
20
#include "infiniop/ops/equal.h"
21
#include "infiniop/ops/flash_attention.h"
22
#include "infiniop/ops/gelu.h"
PanZezhongQY's avatar
PanZezhongQY committed
23
#include "infiniop/ops/gemm.h"
PanZezhong's avatar
PanZezhong committed
24
25
#include "infiniop/ops/hardswish.h"
#include "infiniop/ops/hardtanh.h"
qinyiqun's avatar
qinyiqun committed
26
#include "infiniop/ops/int8_gemm.h"
27
#include "infiniop/ops/kv_caching.h"
28
29
30
#include "infiniop/ops/layer_norm.h"
#include "infiniop/ops/logsoftmax.h"
#include "infiniop/ops/lp_norm.h"
Graylatzhou's avatar
Graylatzhou committed
31
#include "infiniop/ops/mul.h"
32
#include "infiniop/ops/ones.h"
33
34
35
#include "infiniop/ops/paged_attention.h"
#include "infiniop/ops/paged_attention_prefill.h"
#include "infiniop/ops/paged_caching.h"
qinyiqun's avatar
qinyiqun committed
36
#include "infiniop/ops/quant/per_channel_quant_int8.h"
PanZezhongQY's avatar
PanZezhongQY committed
37
38
#include "infiniop/ops/random_sample.h"
#include "infiniop/ops/rearrange.h"
PanZezhong's avatar
PanZezhong committed
39
#include "infiniop/ops/reciprocal.h"
PanZezhongQY's avatar
PanZezhongQY committed
40
41
#include "infiniop/ops/relu.h"
#include "infiniop/ops/rms_norm.h"
42
#include "infiniop/ops/rope.h"
43
#include "infiniop/ops/sigmoid.h"
44
#include "infiniop/ops/silu.h"
45
#include "infiniop/ops/silu_and_mul.h"
46
#include "infiniop/ops/softmax.h"
47
#include "infiniop/ops/softplus.h"
Pepe's avatar
Pepe committed
48
#include "infiniop/ops/sub.h"
PanZezhong's avatar
PanZezhong committed
49
#include "infiniop/ops/sum.h"
PanZezhongQY's avatar
PanZezhongQY committed
50
#include "infiniop/ops/swiglu.h"
51
#include "infiniop/ops/tanh.h"
PanZezhong's avatar
PanZezhong committed
52
#include "infiniop/ops/topk.h"
blkmjsian's avatar
blkmjsian committed
53
#include "infiniop/ops/topkrouter.h"
54
#include "infiniop/ops/topksoftmax.h"
PanZezhong's avatar
PanZezhong committed
55
56
#include "infiniop/ops/var.h"
#include "infiniop/ops/var_mean.h"
57
#include "infiniop/ops/zeros.h"
58
#include "infiniop/tensor_descriptor.h"
PanZezhongQY's avatar
PanZezhongQY committed
59

PanZezhong's avatar
PanZezhong committed
60
#endif // __INFINIOP_API_H__