gpuでnumpy行列をソート

Question

私は第1列に基づいてソートしたい大きな行列 - 1045506 x 3を持っています。 、それはnumpyの行列なので、私は結果

mat_sorted = mat[mat[:,0].argsort()]

を取得するためにargsortを使用することができますそれは少し私には高いように思え、この手順を完了するのに約69msかかります。

私の質問は、GPUを使用して計算を高速化する方法があるかどうかです。私はpycuda、theanoのようないくつかの図書館を見ていましたが、私はドキュメントを理解することが難しいと思っています。特に、numpy配列と行列をgpuに渡す部分。どのポインタも非常に便利です。

Answer 1

私が知っているPyCUDAまたはscikit-cudaには、ネイティブソート機能はありません。しかし、GPUでの効率的な並列ソートはほとんど解決された問題です.CUDA 4.0以降のツールキットのすべてのバージョンに同梱されているThrustテンプレートライブラリでは、最先端のソートアルゴリズムを見つけることができます。

PyCUDA wikiには、PyCUDAのThrustソートアルゴリズムを呼び出すcomplete exampleが含まれています。