Polygraphy unable to switch context to driver api instead of default context from run time api

[yshvrdhn]

Looking for guidance on how to override Polygraphy's default Runtime API CUDA context management to use a custom Driver API context instead. Currently, Polygraphy automatically creates and manages CUDA context using Runtime API, but our custom trt plugin requires integration with specific Driver API context management.

[LeoZDong]

Could you clarify what you mean by this? Are you referring to the CUDA stream and other wrappers defined in cuda.py? If so, there's no out-of-the-box way to swap out those wrappers, but you could modify source and build from source.

Providing a concrete example might help.