Use shared memory for mat vec mul kernel

This commit is contained in:
2024-03-13 22:13:11 +01:00
parent 09480e42e5
commit 77004c16be
4 changed files with 77 additions and 7 deletions

View File

@@ -7,6 +7,7 @@ add_executable(test_main
layers/test_input.cu
kernels/test_activations.cu
kernels/test_padding.cu
kernels/test_matmul.cu
)
target_link_libraries(test_main ${GTEST_BOTH_LIBRARIES} CUDANet)