Skip to content
This repository has been archived by the owner on Jun 27, 2024. It is now read-only.

[cuDNN] cuDNN module should create a dedicated CUDA stream #138

Open
ezhulenev opened this issue Jun 14, 2023 · 0 comments
Open

[cuDNN] cuDNN module should create a dedicated CUDA stream #138

ezhulenev opened this issue Jun 14, 2023 · 0 comments
Labels
enhancement New feature or request

Comments

@ezhulenev
Copy link
Contributor

Right now all cuDNN operations launched on a NULL stream. CudnnModule at initialization time should create a CUDA stream that it will use for running all operations. Maybe even a pool of CUDA streams?
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ezhulenev