#ifndef __INFINIOP_API_H__ #define __INFINIOP_API_H__ #include "infiniop/handle.h" #include "infiniop/ops/add.h" #include "infiniop/ops/add_rms_norm.h" #include "infiniop/ops/all.h" #include "infiniop/ops/attention.h" #include "infiniop/ops/causal_softmax.h" #include "infiniop/ops/clip.h" #include "infiniop/ops/conv.h" #include "infiniop/ops/dequantize_awq.h" #include "infiniop/ops/embedding.h" #include "infiniop/ops/flash_attention.h" #include "infiniop/ops/gelu.h" #include "infiniop/ops/gemm.h" #include "infiniop/ops/int8_gemm.h" #include "infiniop/ops/kv_caching.h" #include "infiniop/ops/layer_norm.h" #include "infiniop/ops/logsoftmax.h" #include "infiniop/ops/lp_norm.h" #include "infiniop/ops/mul.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/quant/per_channel_quant_int8.h" #include "infiniop/ops/random_sample.h" #include "infiniop/ops/rearrange.h" #include "infiniop/ops/relu.h" #include "infiniop/ops/rms_norm.h" #include "infiniop/ops/rope.h" #include "infiniop/ops/sigmoid.h" #include "infiniop/ops/silu.h" #include "infiniop/ops/silu_and_mul.h" #include "infiniop/ops/softmax.h" #include "infiniop/ops/softplus.h" #include "infiniop/ops/sub.h" #include "infiniop/ops/sum.h" #include "infiniop/ops/swiglu.h" #include "infiniop/ops/tanh.h" #include "infiniop/ops/topk.h" #include "infiniop/ops/topkrouter.h" #include "infiniop/ops/topksoftmax.h" #include "infiniop/ops/var.h" #include "infiniop/ops/var_mean.h" #include "infiniop/ops/zeros.h" #include "infiniop/tensor_descriptor.h" #include "infiniop/ops/cross_entropy.h" #include "infiniop/ops/hardswish.h" #include "infiniop/ops/avg_pool1d.h" #include "infiniop/ops/equal.h" #include "infiniop/ops/hardtanh.h" #endif // __INFINIOP_API_H__