Open ehsantn opened 7 years ago
We need a more general form of prange that provides a parallel iterator in multiple dimensions. Currently, parallel functions like min/max have to linearize the array using ravel first.
prange
ravel
There is a ndindex function in numpy that iterates over indices, which is actually very close to our loopnest. So maybe some like a pndindex?
ndindex
pndindex
Good idea.
Yes, this looks great for our purpose.
We need a more general form of
prange
that provides a parallel iterator in multiple dimensions. Currently, parallel functions like min/max have to linearize the array usingravel
first.