bartorch.set_cuda_streams

bartorch.set_cuda_streams#

bartorch.set_cuda_streams(n)#

Set the number of CUDA streams BART runs on, 1 to 8.

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