Set limit on n_streams (Issue 2076)

Fixes #2076 .
You would have to specify `dtype=float32` for it to actually be computed on the GPU, otherwise the assertion fails.