Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement device vector utils

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-11 22:22:33 +02:00
parent 710a33bdde
commit 4b9d123e94
6 changed files with 109 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/kernels/test_matmul.cu
View File

@@ -4,6 +4,7 @@
#include <vector>
#include "cuda_helper.cuh"
#include "vector.cuh"
#include "matmul.cuh"
TEST(MatMulTest, MatVecMulTest) {
@@ -45,7 +46,7 @@ TEST(MatMulTest, MatVecMulTest) {
int THREADS_PER_BLOCK = std::max(w, h);
int BLOCKS = 1;
CUDANet::Kernels::clear<<<BLOCKS, h>>>(d_output, h);
CUDANet::Utils::clear(d_output, h);
CUDANet::Kernels::mat_vec_mul<<<BLOCKS, THREADS_PER_BLOCK, sizeof(float) * w>>>(d_matrix, d_vector, d_output, w, h);
cudaStatus = cudaDeviceSynchronize();
@@ -198,6 +199,7 @@ TEST(MatMulTest, SumReduceTest) {
remaining = blocks_needed;
}
std::vector<float> sum(n);
cudaStatus = cudaMemcpy(
sum.data(), d_sum, sizeof(float) * n, cudaMemcpyDeviceToHost