zdevito / ATen

ATen: A TENsor library for C++11
683 stars 125 forks source link

Cuda Streams #227

Open soulslicer opened 5 years ago

soulslicer commented 5 years ago

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