"googlemock/docs/under-construction-banner.md" did not exist on "389cb68b87193358358ae87cc56d257fd0d80189"
infiniop.h 1.12 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"
PanZezhongQY's avatar
PanZezhongQY committed
7
8
#include "infiniop/ops/attention.h"
#include "infiniop/ops/causal_softmax.h"
goldenfox2025's avatar
goldenfox2025 committed
9
#include "infiniop/ops/clip.h"
PanZezhongQY's avatar
PanZezhongQY committed
10
#include "infiniop/ops/conv.h"
11
#include "infiniop/ops/dequantize_awq.h"
12
#include "infiniop/ops/gelu.h"
PanZezhongQY's avatar
PanZezhongQY committed
13
#include "infiniop/ops/gemm.h"
14
15
16
#include "infiniop/ops/layer_norm.h"
#include "infiniop/ops/logsoftmax.h"
#include "infiniop/ops/lp_norm.h"
Graylatzhou's avatar
Graylatzhou committed
17
#include "infiniop/ops/mul.h"
18
#include "infiniop/ops/ones.h"
PanZezhongQY's avatar
PanZezhongQY committed
19
20
21
22
#include "infiniop/ops/random_sample.h"
#include "infiniop/ops/rearrange.h"
#include "infiniop/ops/relu.h"
#include "infiniop/ops/rms_norm.h"
23
#include "infiniop/ops/rope.h"
24
#include "infiniop/ops/sigmoid.h"
25
#include "infiniop/ops/silu.h"
26
#include "infiniop/ops/softmax.h"
27
#include "infiniop/ops/softplus.h"
Pepe's avatar
Pepe committed
28
#include "infiniop/ops/sub.h"
PanZezhongQY's avatar
PanZezhongQY committed
29
#include "infiniop/ops/swiglu.h"
30
#include "infiniop/ops/tanh.h"
blkmjsian's avatar
blkmjsian committed
31
#include "infiniop/ops/topkrouter.h"
32
#include "infiniop/ops/topksoftmax.h"
33
#include "infiniop/ops/zeros.h"
34
#include "infiniop/tensor_descriptor.h"
PanZezhongQY's avatar
PanZezhongQY committed
35

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