ATen
ATen copied to clipboard
Cuda Streams
How does one "select" which cuda stream to run on. Is there a way to pass that into the functions some how?
How does one "select" which cuda stream to run on. Is there a way to pass that into the functions some how?