Merge branch 'main' into 'dev'
Integration of I-ViT operators See merge request !34
No related branches found
No related tags found
Showing
- include/aidge/backend/cuda.hpp 5 additions, 0 deletionsinclude/aidge/backend/cuda.hpp
- include/aidge/backend/cuda/operator/ILayerNormImpl.hpp 65 additions, 0 deletionsinclude/aidge/backend/cuda/operator/ILayerNormImpl.hpp
- include/aidge/backend/cuda/operator/ILayerNormImpl_CUDA_kernels.hpp 92 additions, 0 deletions...dge/backend/cuda/operator/ILayerNormImpl_CUDA_kernels.hpp
- include/aidge/backend/cuda/operator/ShiftGELUImpl.hpp 7 additions, 3 deletionsinclude/aidge/backend/cuda/operator/ShiftGELUImpl.hpp
- include/aidge/backend/cuda/operator/ShiftGELUImpl_CUDA_kernels.hpp 47 additions, 3 deletions...idge/backend/cuda/operator/ShiftGELUImpl_CUDA_kernels.hpp
- include/aidge/backend/cuda/operator/ShiftMaxImpl.hpp 6 additions, 3 deletionsinclude/aidge/backend/cuda/operator/ShiftMaxImpl.hpp
- include/aidge/backend/cuda/operator/ShiftMaxImpl_CUDA_kernels.hpp 48 additions, 3 deletions...aidge/backend/cuda/operator/ShiftMaxImpl_CUDA_kernels.hpp
- src/operator/ILayerNormImpl.cpp 204 additions, 0 deletionssrc/operator/ILayerNormImpl.cpp
- src/operator/ILayerNormImpl_CUDA_kernels.cu 335 additions, 0 deletionssrc/operator/ILayerNormImpl_CUDA_kernels.cu
- src/operator/ShiftGELUImpl.cpp 38 additions, 15 deletionssrc/operator/ShiftGELUImpl.cpp
- src/operator/ShiftGELUImpl_CUDA_kernels.cu 163 additions, 54 deletionssrc/operator/ShiftGELUImpl_CUDA_kernels.cu
- src/operator/ShiftMaxImpl.cpp 40 additions, 15 deletionssrc/operator/ShiftMaxImpl.cpp
- src/operator/ShiftMaxImpl_CUDA_kernels.cu 203 additions, 69 deletionssrc/operator/ShiftMaxImpl_CUDA_kernels.cu
- unit_tests/Test_ILayerNormImpl.cpp 201 additions, 0 deletionsunit_tests/Test_ILayerNormImpl.cpp
- unit_tests/Test_ShiftGELUImpl.cpp 92 additions, 3 deletionsunit_tests/Test_ShiftGELUImpl.cpp
- unit_tests/Test_ShiftMaxImpl.cpp 91 additions, 2 deletionsunit_tests/Test_ShiftMaxImpl.cpp
Loading
Please register or sign in to comment