about summary refs log tree commit diff
path: root/pkgs/games
diff options
context:
space:
mode:
authorOmnipotentEntity <OmnipotentEntity@gmail.com>2024-03-05 14:54:48 -0500
committerGitHub <noreply@github.com>2024-03-05 20:54:48 +0100
commit6a04adccdddf869eeff79a01c02fb11a3f0e0bf9 (patch)
tree8ec73e9165546b7e9de6bb633b60cb2a36c341cd /pkgs/games
parent6dc8cbe3cc1520315d85c3e4490b50a73c7c7381 (diff)
downloadnixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar.gz
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar.bz2
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar.lz
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar.xz
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.tar.zst
nixlib-6a04adccdddf869eeff79a01c02fb11a3f0e0bf9.zip
katago: expose CMake options new in 1.14.0 (#277390)
Diffstat (limited to 'pkgs/games')
-rw-r--r--pkgs/games/katago/default.nix27
1 files changed, 9 insertions, 18 deletions
diff --git a/pkgs/games/katago/default.nix b/pkgs/games/katago/default.nix
index 4a66f6cc711e..834ec0774d7a 100644
--- a/pkgs/games/katago/default.nix
+++ b/pkgs/games/katago/default.nix
@@ -19,6 +19,7 @@
 , enableBigBoards ? false
 , enableContrib ? false
 , enableTcmalloc ? true
+, enableTrtPlanCache ? false
 }:
 
 assert lib.assertOneOf "backend" backend [ "opencl" "cuda" "tensorrt" "eigen" ];
@@ -67,25 +68,15 @@ stdenv.mkDerivation rec {
   ];
 
   cmakeFlags = [
-    "-DNO_GIT_REVISION=ON"
-  ] ++ lib.optionals enableAVX2 [
-    "-DUSE_AVX2=ON"
-  ] ++ lib.optionals (backend == "eigen") [
-    "-DUSE_BACKEND=EIGEN"
-  ] ++ lib.optionals (backend == "cuda") [
-    "-DUSE_BACKEND=CUDA"
-  ] ++ lib.optionals (backend == "tensorrt") [
-    "-DUSE_BACKEND=TENSORRT"
-  ] ++ lib.optionals (backend == "opencl") [
-    "-DUSE_BACKEND=OPENCL"
+    (lib.cmakeFeature "USE_BACKEND" (lib.toUpper backend))
+    (lib.cmakeBool "USE_AVX2" enableAVX2)
+    (lib.cmakeBool "USE_TCMALLOC" enableTcmalloc)
+    (lib.cmakeBool "USE_BIGGER_BOARDS_EXPENSIVE" enableBigBoards)
+    (lib.cmakeBool "USE_CACHE_TENSORRT_PLAN" enableTrtPlanCache)
+    (lib.cmakeBool "NO_GIT_REVISION" (!enableContrib))
   ] ++ lib.optionals enableContrib [
-    "-DBUILD_DISTRIBUTED=1"
-    "-DNO_GIT_REVISION=OFF"
-    "-DGIT_EXECUTABLE=${fakegit}/bin/git"
-  ] ++ lib.optionals enableTcmalloc [
-    "-DUSE_TCMALLOC=ON"
-  ] ++ lib.optionals enableBigBoards [
-    "-DUSE_BIGGER_BOARDS_EXPENSIVE=ON"
+    (lib.cmakeBool "BUILD_DISTRIBUTED" true)
+    (lib.cmakeFeature "GIT_EXECUTABLE" "${fakegit}/bin/git")
   ];
 
   preConfigure = ''