Set limit on n_streams (Issue 2076)

Fixes #2076 .
Can you make a test for the new back-end to make sure that it work correctly with those inputs? Maybe you can update the current tests (grep for GPUA) to test the new shapes case.