From 246aeb9c20c63d258434500a76dd78157d4b6b86 Mon Sep 17 00:00:00 2001
From: NevermindNilas <nilascontact@gmail.com>
Date: Tue, 7 May 2024 01:26:42 +0300
Subject: [PATCH] update things

---
 .github/workflows/build.yaml | 8 ++++++++
 build.py                     | 7 +++++++
 requirements.txt             | 1 +
 3 files changed, 16 insertions(+)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 691aa10f..1d835b5d 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -56,6 +56,14 @@ jobs:
         run: |
           echo "Version: $VERSION"
 
+      - name: Git Clone TensorRT
+        run: git clone https://github.com/NVIDIA/TensorRT
+
+      - name: CD To Polygraphy and Install Latest Polygraphy
+        run: |
+          cd TensorRT/tools/Polygraphy
+          ./install.ps1
+
       - name: Create Release
         id: create_release
         uses: actions/create-release@v1
diff --git a/build.py b/build.py
index 7f8b8535..5ec44c5f 100644
--- a/build.py
+++ b/build.py
@@ -27,6 +27,13 @@ def install_requirements():
         check=True,
     )
 
+def install_polygraphy():
+    print("Installing Polygraphy...")
+    cwd = os.getcwd()
+    os.chdir("TensorRT/Tools/Polygraphy")
+    subprocess.run(["powershell", "-File", "install.ps1"], check=True)
+    os.chdir(cwd)
+
 
 def install_pyinstaller():
     print("Installing PyInstaller...")
diff --git a/requirements.txt b/requirements.txt
index 2939ef50..06efc969 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,6 +12,7 @@ spandrel
 yt-dlp
 requests
 tqdm
+tensorrt
 colorama
 einops
 scikit-image