feat : add ViT operators
Context
This MR intends to add missing operators implementation to support ViT model inference:
- Erf
- MatMul
- Softmax
It handles issue #27
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 -
MatMul -
Softmax
Edited by Houssem ROUIS