From 05146f93fee979c42838ca37ffd72dab295d8067 Mon Sep 17 00:00:00 2001
From: "Micah Chambers (eos)" <mchambers@anduril.com>
Date: Wed, 6 Aug 2025 10:23:59 -0700
Subject: [PATCH] add version

---
 src/ffi/mod.rs     |  1 +
 src/ffi/version.rs | 16 ++++++++++++++++
 2 files changed, 17 insertions(+)
 create mode 100644 src/ffi/version.rs

diff --git a/src/ffi/mod.rs b/src/ffi/mod.rs
index 30651ce..72fdccc 100644
--- a/src/ffi/mod.rs
+++ b/src/ffi/mod.rs
@@ -16,6 +16,7 @@ pub mod network;
 pub mod optimization_profile;
 pub mod parser;
 pub mod sync;
+pub mod version;
 
 /// Convenience macro for turning TensorRT error code into a `std::result::Result`.
 ///
diff --git a/src/ffi/version.rs b/src/ffi/version.rs
new file mode 100644
index 0000000..5a037a2
--- /dev/null
+++ b/src/ffi/version.rs
@@ -0,0 +1,16 @@
+use cpp::cpp;
+
+/// Returns (Major, Minor, Patch, Build) version of tensorrt
+pub fn get_tensorrt_version() -> (u32, u32, u32) {
+    (
+        cpp!(unsafe [] -> u32 as "uint32_t" {
+            return NV_TENSORRT_MAJOR;
+        }),
+        cpp!(unsafe [] -> u32 as "uint32_t" {
+            return NV_TENSORRT_MINOR;
+        }),
+        cpp!(unsafe [] -> u32 as "uint32_t" {
+            return NV_TENSORRT_PATCH;
+        }),
+    )
+}