#cuda #gpu #async #npp #nvidia

async-cuda-npp

Async NVIDIA Performance Primitives for Rust

6 releases (3 breaking)

0.4.0 Aug 10, 2023
0.3.0 Aug 8, 2023
0.2.0 Aug 8, 2023
0.1.3 Jun 15, 2023

#1203 in Asynchronous

MIT/Apache

200KB
4K SLoC

async-cuda-npp

Asynchronous wrapper for the NVIDIA 2D Image And Signal Performance Primitives library.

version license docs

ℹ️ Introduction

async-cuda-npp is an async wrapper for NVIDIA's NPP library. It is part of the async-cuda family of libraries.

🛠 S️️tatus

This project is still a work-in-progress, and will contain bugs. Some parts of the API have not been flushed out yet. Use with caution.

📦 Setup

Make sure you have the necessary dependencies installed:

  • CUDA toolkit 11 or later (NPP is part of the CUDA toolkit).

Then, add the following to your dependencies in Cargo.toml:

async-cuda-npp = "0.1"

⚠️ Safety warning

This crate is intentionally unsafe. Due to the limitations of how async Rust currently works, usage of the async interface of this crate can cause undefined behavior in some rare cases. It is up to the user of this crate to prevent this from happening by following these rules:

  • No futures produced by functions in this crate may be leaked (either by std::mem::forget or otherwise).
  • Use a well-behaved runtime (one that will not forget your future) like Tokio or async-std.

Internally, the Future type in this crate schedules a CUDA call on a separate runtime thread. To make the API as ergonomic as possible, the lifetime bounds of the closure (that is sent to the runtime) are tied to the future object. To enforce this bound, the future will block and wait if it is dropped. This mechanism relies on the future being driven to completion, and not forgotten. This is not necessarily guaranteed. Unsafety may arise if either the runtime gives up on or forgets the future, or the caller manually polls the future, then forgets it.

License

Licensed under either of

at your option.

Contribution

Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.

Dependencies

~1–2.2MB
~41K SLoC