| 16 | // transpose a matrix |
| 17 | template <typename T> |
| 18 | void transpose(T* out, const T* in, size_s width, size_s height) { |
| 19 | dim3 blockDim(32, 32); |
| 20 | dim3 gridDim((width + blockDim.x - 1) / blockDim.x, (height + blockDim.y - 1) / blockDim.y); |
| 21 | transposeKernel<<<gridDim, blockDim>>>(out, in, width, height); |
| 22 | } |
| 23 | |
| 24 | template <typename T> |
| 25 | __global__ void symKernel(T* out, const T* in, int mat_size) { |