WIP: conv impl with CuDNN
Showing
- .gitlab/ci/_global.gitlab-ci.yml 1 addition, 0 deletions.gitlab/ci/_global.gitlab-ci.yml
- CMakeLists.txt 1 addition, 0 deletionsCMakeLists.txt
- include/aidge/backend/cuda/data/TensorImpl.hpp 48 additions, 0 deletionsinclude/aidge/backend/cuda/data/TensorImpl.hpp
- include/aidge/backend/cuda/operator/ConvImpl.hpp 76 additions, 0 deletionsinclude/aidge/backend/cuda/operator/ConvImpl.hpp
- include/aidge/backend/cuda/utils/CudaContext.hpp 165 additions, 0 deletionsinclude/aidge/backend/cuda/utils/CudaContext.hpp
- include/aidge/backend/cuda/utils/CudaUtils.hpp 78 additions, 10 deletionsinclude/aidge/backend/cuda/utils/CudaUtils.hpp
- src/operator/ConvImpl.cpp 189 additions, 0 deletionssrc/operator/ConvImpl.cpp
- src/utils/CudaUtils.cpp 51 additions, 0 deletionssrc/utils/CudaUtils.cpp
Loading
Please register or sign in to comment