bartorch.set_cuda_streams

bartorch.set_cuda_streams()

Set the number of CUDA streams BART runs on.

More than one lets BART overlap its transfers with its arithmetic.

Parameters:

n (int) – Stream count, 1 to 8.

Raises:

ValueError – If BART refuses n.