Skip to content
Snippets Groups Projects

feat : add ViT operators

Merged 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
  • MatMul
  • Softmax

It handles issue #27 (moved)

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
  • MatMul
  • Softmax
Edited by Houssem ROUIS

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
  • Loading
Please register or sign in to reply
Loading