将 cudaStream_t 对象转换为 CUStream 对象

Question

有什么方法可以将 cudaStream_t 对象转换为 CUStream 吗？

我找到了 this hack，但我认为使用它不安全。 或者，是否有任何方法可以仅在 CUDA 7.5 中使用 cudaStream_t 对象在 JCuda 应用程序中调用 cudaLaunchKernel？

Answer 1

在运行时和驱动程序 APIs 中的 C 级别，cudaStream_t 和 CUStream 是 same type，可以在 API 中互换使用].

在 JCUDA 级别，CUstream 似乎有一个 specialization 的构造函数，用于使用现有 cudaStream_t 实例初始化实例。这个替代构造函数的提供消除了您在问题中链接到的指针交换的需要。