Skip to content

Add stream pool to avoid cuStreamCreate overhead #2012

Open
@simonbyrne

Description

@simonbyrne

Describe the bug

I tried adding multithreading to our code, and now we see occasional pauses due to CuStreamCreate:
Screenshot 2023-07-29 at 5 27 06 PM

Could be fixed by adding a Stream pool?

To reproduce

Available on request

Expected behavior

No pauses

Version info

Not sure exactly, I think it was Julia 1.8.5 with CUDA.jl 3.13.1 and CUDA 11.2.

Once our cluster upgrades the drivers, I can rerun with CUDA 12.

Additional context

cc @lcw

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions