infiniop.h 795 Bytes
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
6
7
8
#include "infiniop/ops/add.h"
#include "infiniop/ops/attention.h"
#include "infiniop/ops/avg_pool.h"
#include "infiniop/ops/causal_softmax.h"
goldenfox2025's avatar
goldenfox2025 committed
9
#include "infiniop/ops/clip.h"
PanZezhongQY's avatar
PanZezhongQY committed
10
11
12
13
14
15
#include "infiniop/ops/conv.h"
#include "infiniop/ops/expand.h"
#include "infiniop/ops/gemm.h"
#include "infiniop/ops/global_avg_pool.h"
#include "infiniop/ops/max_pool.h"
#include "infiniop/ops/mlp.h"
Graylatzhou's avatar
Graylatzhou committed
16
#include "infiniop/ops/mul.h"
PanZezhongQY's avatar
PanZezhongQY committed
17
18
19
20
#include "infiniop/ops/random_sample.h"
#include "infiniop/ops/rearrange.h"
#include "infiniop/ops/relu.h"
#include "infiniop/ops/rms_norm.h"
21
#include "infiniop/ops/rope.h"
Pepe's avatar
Pepe committed
22
#include "infiniop/ops/sub.h"
PanZezhongQY's avatar
PanZezhongQY committed
23
#include "infiniop/ops/swiglu.h"
24
#include "infiniop/tensor_descriptor.h"
PanZezhongQY's avatar
PanZezhongQY committed
25

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