Skip to content

Vit operators

Houssem ROUIS requested to merge hrouis/aidge_backend_cuda:vit_operators into dev

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 and Test_MatMulImpl.cpp, Add MatMul operator;
  • ErfImpl.hpp, ErfImpl.cpp, ErfImpl_CUDA_kernels.hpp, ErfImpl_CUDA_kernels.cu and Test_ErfImpl.cpp, Add Erf operator;
  • SoftmaxImpl.hpp, SoftmaxImpl.cpp and Test_SoftmaxImpl.cpp, Add Softmax operator;

TODO

  • Erf
  • Gather
  • MatMul
  • Slice
  • Softmax

Merge request reports

Loading