提交 14cf7b8d authored 作者: Tim Cooijmans's avatar Tim Cooijmans

GpuBatchedDot: switch back to NULL stream

上级 e54fcb5d
......@@ -295,6 +295,7 @@ class GpuBatchedDot(GpuOp):
x += Sx[0]; y += Sy[0]; z += Sz[0];
};
cublasSetStream(handle, NULL);
for (int i = 0; i < N_STREAMS; i++) {
cudaStreamSynchronize(streams[i]);
cudaStreamDestroy(streams[i]);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论