use tracing to automatically infer static and dynamic shapes as well as which modules need to be compiled in nested pipelies objects, then compile with Torch-TensorRT
-
Notifications
You must be signed in to change notification settings - Fork 0
automatically infer dynamic shapes for Torch-Tensor compilation
License
replicate/autocompile
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
automatically infer dynamic shapes for Torch-Tensor compilation
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published