From 36fada3a6716c948ce7a5ace4d8b6f7021b01296 Mon Sep 17 00:00:00 2001
From: Connor Baker <ConnorBaker01@gmail.com>
Date: Tue, 25 Nov 2025 14:56:13 +0000
Subject: [PATCH] katago: patch outdated TensorRT version detection

Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
---
 pkgs/by-name/ka/katago/package.nix | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkgs/by-name/ka/katago/package.nix b/pkgs/by-name/ka/katago/package.nix
index d580be37282d..7bc1f7f30024 100644
--- a/pkgs/by-name/ka/katago/package.nix
+++ b/pkgs/by-name/ka/katago/package.nix
@@ -61,6 +61,17 @@ stdenv'.mkDerivation rec {
     makeWrapper
   ];
 
+  # Included from release 1.16.2:
+  # https://github.com/lightvector/KataGo/commit/9030f72d152da42c1dd03590aa5116993ea842f6
+  # Doesn't apply cleanly as a patch so doing a quick replacement to the same effect.
+  prePatch = lib.optionalString (backend == "tensorrt") ''
+    nixLog "patching $PWD/cpp/CMakeLists.txt to work around outdated TensorRT version detection"
+    substituteInPlace "$PWD/cpp/CMakeLists.txt" \
+      --replace-fail \
+        'if(TENSORRT_VERSION VERSION_LESS 8.5)' \
+        'if(NOT TENSORRT_VERSION STREQUAL ".." AND TENSORRT_VERSION VERSION_LESS 8.5)'
+  '';
+
   buildInputs = [
     libzip
     boost