Vit operators
Context
This MR intends to add missing operators implementation to support ViT model inference:
- Erf
- Gather
- MatMul
- Slice
- Softmax
Modified files
-
MatMulImpl.hpp
,MatMulImpl.cpp
andTest_MatMulImpl.cpp
, Add MatMul operator; -
ErfImpl.hpp
,ErfImpl.cpp
,ErfImpl_CUDA_kernels.hpp
,ErfImpl_CUDA_kernels.cu
andTest_ErfImpl.cpp
, Add Erf operator; -
SoftmaxImpl.hpp
,SoftmaxImpl.cpp
andTest_SoftmaxImpl.cpp
, Add Softmax operator;
TODO
-
Erf -
Gather -
MatMul -
Slice -
Softmax