From 74ff9ebf966079597ec8923534c9ab93874758c3 Mon Sep 17 00:00:00 2001
From: Violet <c01368481@gmail.com>
Date: Mon, 23 Jun 2025 16:14:07 -0700
Subject: [PATCH] Remove trailing zeroes from end of ptx (#390)

---
 dark_api/src/fatbin.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/dark_api/src/fatbin.rs b/dark_api/src/fatbin.rs
index 4e583c8..b14a79f 100644
--- a/dark_api/src/fatbin.rs
+++ b/dark_api/src/fatbin.rs
@@ -142,7 +142,7 @@ impl<'a> FatbinFile<'a> {
     }
 
     pub unsafe fn decompress(&'a self) -> Result<Vec<u8>, FatbinError> {
-        let payload = if self
+        let mut payload = if self
             .header
             .flags
             .contains(FatbinFileHeaderFlags::CompressedLz4)
@@ -158,6 +158,12 @@ impl<'a> FatbinFile<'a> {
             unsafe { self.get_payload().to_vec() }
         };
 
+        
+        while payload.last() == Some(&0) {
+            // remove trailing zeros
+            payload.pop();
+        }
+
         Ok(payload)
     }
 }