From 1c276d7d19518eabac2454bbbcee80dadbec8c95 Mon Sep 17 00:00:00 2001
From: Chang Sun <chsun@cern.ch>
Date: Tue, 20 Feb 2024 17:52:19 -0800
Subject: [PATCH] vitis backend tarball fix

---
 hls4ml/writer/vitis_writer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/hls4ml/writer/vitis_writer.py b/hls4ml/writer/vitis_writer.py
index 106cdbf1c6..cfb4c0cb27 100644
--- a/hls4ml/writer/vitis_writer.py
+++ b/hls4ml/writer/vitis_writer.py
@@ -30,3 +30,5 @@ def write_hls(self, model):
         """
         super().write_hls(model)
         self.write_nnet_utils_overrides(model)
+        os.remove(model.config.get_output_dir() + '.tar.gz')
+        self.write_tar(model)