Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.hpp 257 Bytes
Newer Older
PanZezhong1725's avatar
issue/461 InfiniCore 推理运行时  
PanZezhong1725 committed
1
2 
#pragma once
thatPepe's avatar
Issue/497 - Enhanced Test Framework (#520)  
thatPepe committed
3
4 
#include "ops/add.hpp"
#include "ops/attention.hpp"
zhuyue's avatar
Add causalSoftmax operator Python interface and tests.  
zhuyue committed
5 
#include "ops/causal_softmax.hpp"
thatPepe's avatar
Issue/497 - Enhanced Test Framework (#520)  
thatPepe committed
6
7
8
9 
#include "ops/matmul.hpp"
#include "ops/ones.hpp"
#include "ops/rearrange.hpp"
#include "ops/rms_norm.hpp"
zhuyue's avatar
Add silu operator python interface and tests.  
zhuyue committed
10 
#include "ops/silu.hpp"
zhuyue's avatar
Add SwiGLU operator Python interface and tests.  
zhuyue committed
11 
#include "ops/swiglu.hpp"