就地CUDA內核長方形矩陣轉置

Question

我已經仔細研讀了一段時間，但無法找到這個正確的答案：

是否有實現就地對角矩陣轉置的CUDA ？

我知道cublas geam，但這需要創建另一個矩陣。我嘗試了一個天真的執行：CUDA In-place Transpose Error

但是，這隻適用於矩陣矩陣。有人可以向我解釋爲什麼這個邏輯對於對角矩陣不起作用嗎？雖然「天真」換位的方法可行，但它不適用。

Answer 1

環視後一會我發現下面的GitHub的頁面裏面確實有相關的NVIDIA研究論文就地轉碼：

https://github.com/BryanCatanzaro/inplace

這似乎是解決此問題的正確方法。

Answer 2

看一看以下紙張：A Decomposition for In-place Matrix Transposition

序貫算法用於就地矩陣轉置如下（> O（n * m個）運行時間）：

// in: n rows; m cols 
// out: n cols; m rows 
void matrix_transpose(int *a, int n, int m) { 
    int i, j; 
    for(int k = 0; k < n*m; k++) { 
     int idx = k; 
     do { // calculate index in the original array 
      idx = (idx % n) * m + (idx/n); 
     } while(idx < k); // make sure we don't swap elements twice 
     std::swap(a[k], a[idx]); 
    } 
}