@glaringlee
Thank you, Xinyun. I am looking at this particular line of code:
https://github.com/pytorch/extension-cpp/blob/master/cuda/lltm_cuda_kernel.cu#L119 9.
I don’t think CUDA stream was wrapped up there?
Best,
Lei