colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

katago: expose CMake options new in 1.14.0 (#277390)

Browse Source
This commit is contained in:
OmnipotentEntity 2024-03-05 14:54:48 -05:00 committed by GitHub
parent 6dc8cbe3cc
commit 6a04adccdd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pkgs/games/katago/default.nix
View File

@ -19,6 +19,7 @@
, enableBigBoards ? false , enableBigBoards ? false
, enableContrib ? false , enableContrib ? false
, enableTcmalloc ? true , enableTcmalloc ? true
, enableTrtPlanCache ? false
}: }:
assert lib.assertOneOf "backend" backend [ "opencl" "cuda" "tensorrt" "eigen" ]; assert lib.assertOneOf "backend" backend [ "opencl" "cuda" "tensorrt" "eigen" ];
@ -67,25 +68,15 @@ stdenv.mkDerivation rec {
]; ];
cmakeFlags = [ cmakeFlags = [
"-DNO_GIT_REVISION=ON" (lib.cmakeFeature "USE_BACKEND" (lib.toUpper backend))
] ++ lib.optionals enableAVX2 [ (lib.cmakeBool "USE_AVX2" enableAVX2)
"-DUSE_AVX2=ON" (lib.cmakeBool "USE_TCMALLOC" enableTcmalloc)
] ++ lib.optionals (backend == "eigen") [ (lib.cmakeBool "USE_BIGGER_BOARDS_EXPENSIVE" enableBigBoards)
"-DUSE_BACKEND=EIGEN" (lib.cmakeBool "USE_CACHE_TENSORRT_PLAN" enableTrtPlanCache)
] ++ lib.optionals (backend == "cuda") [ (lib.cmakeBool "NO_GIT_REVISION" (!enableContrib))
"-DUSE_BACKEND=CUDA"
] ++ lib.optionals (backend == "tensorrt") [
"-DUSE_BACKEND=TENSORRT"
] ++ lib.optionals (backend == "opencl") [
"-DUSE_BACKEND=OPENCL"
] ++ lib.optionals enableContrib [ ] ++ lib.optionals enableContrib [
"-DBUILD_DISTRIBUTED=1" (lib.cmakeBool "BUILD_DISTRIBUTED" true)
"-DNO_GIT_REVISION=OFF" (lib.cmakeFeature "GIT_EXECUTABLE" "${fakegit}/bin/git")
"-DGIT_EXECUTABLE=${fakegit}/bin/git"
] ++ lib.optionals enableTcmalloc [
"-DUSE_TCMALLOC=ON"
] ++ lib.optionals enableBigBoards [
"-DUSE_BIGGER_BOARDS_EXPENSIVE=ON"
]; ];
preConfigure = '' preConfigure = ''