diff --git a/trtexec-rs/Cargo.toml b/trtexec-rs/Cargo.toml
index 3ece600..c8ae798 100644
--- a/trtexec-rs/Cargo.toml
+++ b/trtexec-rs/Cargo.toml
@@ -33,3 +33,4 @@ serde_json = "1.0"
 [features]
 default = ["rustnn"]
 rustnn = ["dep:rustnn"]
+enterprise = ["rustnn/trtx-enterprise", "trtx/enterprise"]