flatten implemented

2017-06-29 09:24:06 +00:00
parent a04c3eaf88
commit 97226cbc78
7 changed files with 69 additions and 1 deletions
@@ -42,4 +42,15 @@ void resize(int size, value_type **data)
    if (*data != NULL)
        checkCuda( cudaFree(*data) );
    checkCuda( cudaMalloc(data, size*sizeof(value_type)) );
+}
+
+void matrixTranspose(cublasHandle_t handle, value_type* srcData, value_type* dstData, int rows, int cols) {
+
+    value_type *A = srcData, *clone = dstData;
+    int m = rows, n= cols;
+    checkCuda( cudaMemcpy(clone, A, m*n*sizeof(value_type), cudaMemcpyDeviceToDevice));
+   
+    float const alpha(1.0);
+    float const beta(0.0);
+    checkERROR( cublasSgeam( handle, CUBLAS_OP_T, CUBLAS_OP_N, m, n, &alpha, A, n, &beta, A, m, clone, m ));
 }