How to run list comprehensions on GPU?

Question

Ishmael89

2022年5月10日 00:03

Is there a way to run complex list comprehensions like the following on GPU?

[[x[index] if x[index]len(x) else x[index]-1 for x in slice] if (len(slice)==1) else slice for slice,index in zip(slices,indices)]

To what degree is it Possible? Do I have to convert it to some kind of numpy comprehension (if so what part is speciffically possible/necessary)

The goal is performance optimization on large datalists/arrays.

Brian Spiering · Accepted Answer · 2020年10月27日 17:10

Brian Spiering answered at 2020年10月27日 17:10

There are several different methods.

One method to use Numba which is a just-in-time complier. Numba can compile Python to target CUDA. CUDA is a API for GPU computation.