Skip to content

About

TensorRT plugin and CUDA kernels for whIRLwind inference pipelines.

Resources

Stars

1 star

Watchers

0 watching

Forks

Latest commit

 

History

10 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

whirlwind_trt_plugins

TensorRT plugin and CUDA kernels for whIRLwind inference pipelines. The build produces a shared plugin library:

build/libww_trt_plugins.so

Requirements

  • Linux with TensorRT headers/libraries installed under the distro paths (/usr/include/x86_64-linux-gnu and /usr/lib/x86_64-linux-gnu, or their aarch64 equivalents on Orin)
  • CUDA toolkit available at /usr/local/cuda
  • CMake and a C++17 compiler

Build

pixi run build

The Pixi task intentionally uses the host compiler and CUDA toolkit (/usr/bin/c++ and /usr/local/cuda/bin/nvcc) so it matches the manual CMake build.

About

TensorRT plugin and CUDA kernels for whIRLwind inference pipelines.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages