使用cuda挑選數組的元素

Question

我想從滿足特定條件的數組中挑選三角形。 我會如何在CUDA中做類似的事情？輸入是三角形數組。 CUDA然後計算每個三角形的值。 輸出應該是由值小於閾值的那些三角形組成的數組，例如2。

triangle array:  [tri1][tri2][tri3][tri4] 
value array:  [1.00][0.50][3.50][0.50] 
-> pick triangles with value <= 2 
output:    [tri1][tri2][tri4] 

我不知道如何在這種情況下最佳利用cudas並行。

此刻我將數組數組讀回到mathematica並使用Cases []來選擇太慢的三角形。

result = CUDAMemoryGet[cOutput]; 
validTriangleIndices = Cases[Range[Length[result]], x_ /; result[[x]] < threshold]; 

Answer 1

由於@talonmies說，你還沒有真正提供足夠的信息，在你的問題的一個完整的答案。但是，您所描述的內容聽起來很適合Thrust的copy_if算法。檢出this example。