Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement support for CUDA Streams #7

Open
mstallmo opened this issue Apr 29, 2020 · 4 comments
Open

Implement support for CUDA Streams #7

mstallmo opened this issue Apr 29, 2020 · 4 comments
Labels
enhancement New feature or request
Milestone

Comments

@mstallmo
Copy link
Owner

TensorRT engines support two execution modes, sync and async. Sync is already supported and doesn't require any CUDA Streams to run.

For async to be supported we need to have support for creating and passing CUDA streams to the execution context along with the data to be executed.

We may be able to get this support via https://github.com/bheisler/RustaCUDA since it's already wrapping the CUDA API.

@mstallmo mstallmo added enhancement New feature or request help wanted Extra attention is needed labels May 20, 2020
@mstallmo mstallmo added this to the 0.1 milestone May 20, 2020
@mstallmo mstallmo removed the help wanted Extra attention is needed label Oct 20, 2020
@mstallmo
Copy link
Owner Author

Picking up work on this.

@enfipy
Copy link

enfipy commented Nov 19, 2020

The overall project is looking awesome! What is the state of this and is it ready for simple AIs?

@mstallmo
Copy link
Owner Author

@enfipy Thanks! It is ready for simple Als as long as they don't require custom plugins to run in TRT. Work for adding plugin support is on going but might take a little bit since managing the C++ -> Rust -> C++ relationship is kind of tricky.

There are some pretty big QoL changes in the upcoming 0.4 release that I'm still putting the finishing touches on. If you're comfortable with it I would recommend use the crate directly from the develop branch instead of the current published version on crates.io.

Let me know if you run into any issues or sharp edges while using the library!

@enfipy
Copy link

enfipy commented Nov 19, 2020

Brilliant! Thanks a lot for your hard work, I will definitely try this out!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants