commit fadb458e80b1c7c22a017f44acdaec2d326044dd · pyrox.dev/nixpkgs

+46 -23

pkgs/development/python-modules/torch/source/default.nix

···

       3
       3
        
         lib,

     

       4
       4
        
         fetchFromGitHub,

     

       5
       5
        
         fetchFromGitLab,

     

       6
       6
       +
         fetchpatch2,

     

       6
       7
        
         git-unroll,

     

       7
       8
        
         buildPythonPackage,

     

       8
       9
        
         python,

     
···

       41
       42
        
       

     

       42
       43
        
         # Build inputs

     

       43
       44
        
         apple-sdk_13,

     

       45
       45
       +
         openssl,

     

       44
       46
        
         numactl,

     

       45
       47
        
         llvmPackages,

     

       46
       48
        
       

     
···

       116
       118
        
       

     

       117
       119
        
         setBool = v: if v then "1" else "0";

     

       118
       120
        
       

     

       119
       119
       -
         # https://github.com/pytorch/pytorch/blob/v2.7.0/torch/utils/cpp_extension.py#L2343-L2345

     

       121
       121
       +
         # https://github.com/pytorch/pytorch/blob/v2.8.0/torch/utils/cpp_extension.py#L2411-L2414

     

       120
       122
        
         supportedTorchCudaCapabilities =

     

       121
       123
        
           let

     

       122
       124
        
             real = [

     
···

       142
       144
        
               "10.0a"

     

       143
       145
        
               "10.1"

     

       144
       146
        
               "10.1a"

     

       147
       147
       +
               "10.3"

     

       148
       148
       +
               "10.3a"

     

       145
       149
        
               "12.0"

     

       146
       150
        
               "12.0a"

     

       151
       151
       +
               "12.1"

     

       152
       152
       +
               "12.1a"

     

       147
       153
        
             ];

     

       148
       154
        
             ptx = lists.map (x: "${x}+PTX") real;

     

       149
       155
        
           in

     
···

       216
       222
        
             rocm-comgr

     

       217
       223
        
             rocm-device-libs

     

       218
       224
        
             rocm-runtime

     

       225
       225
       +
             rocm-smi

     

       219
       226
        
             clr.icd

     

       220
       227
        
             hipify

     

       221
       228
        
           ];

     
···

       263
       270
        
       buildPythonPackage rec {

     

       264
       271
        
         pname = "torch";

     

       265
       272
        
         # Don't forget to update torch-bin to the same version.

     

       266
       266
       -
         version = "2.7.1";

     

       273
       273
       +
         version = "2.8.0";

     

       267
       274
        
         pyproject = true;

     

       268
       275
        
       

     

       269
       276
        
         stdenv = stdenv';

     
···

       287
       294
        
       

     

       288
       295
        
         patches = [

     

       289
       296
        
           ./clang19-template-warning.patch

     

       297
       297
       +
       

     

       298
       298
       +
           # Do not override PYTHONPATH, otherwise, the build fails with:

     

       299
       299
       +
           # ModuleNotFoundError: No module named 'typing_extensions'

     

       300
       300
       +
           (fetchpatch2 {

     

       301
       301
       +
             name = "cmake-build-preserve-PYTHONPATH";

     

       302
       302
       +
             url = "https://github.com/pytorch/pytorch/commit/231c72240d80091f099c95e326d3600cba866eee.patch";

     

       303
       303
       +
             hash = "sha256-7f9qtr7ljqjKdj3DqpcCTS0/Hr2AwOzOl/HA88yU3zI=";

     

       304
       304
       +
           })

     

       290
       305
        
         ]

     

       291
       291
       -
         ++ lib.optionals cudaSupport [ ./fix-cmake-cuda-toolkit.patch ]

     

       306
       306
       +
         ++ lib.optionals cudaSupport [

     

       307
       307
       +
           ./fix-cmake-cuda-toolkit.patch

     

       308
       308
       +
           ./nvtx3-hpp-path-fix.patch

     

       309
       309
       +
         ]

     

       292
       310
        
         ++ lib.optionals stdenv.hostPlatform.isLinux [

     

       293
       311
        
           # Propagate CUPTI to Kineto by overriding the search path with environment variables.

     

       294
       312
        
           # https://github.com/pytorch/pytorch/pull/108847

     
···

       302
       320
        
         ];

     

       303
       321
        
       

     

       304
       322
        
         postPatch = ''

     

       305
       305
       -
           # Prevent NCCL from being cloned during the configure phase

     

       306
       306
       -
           # TODO: remove when updating to the next release as it will not be needed anymore

     

       307
       307
       -
           substituteInPlace tools/build_pytorch_libs.py \

     

       308
       308
       -
             --replace-fail "  checkout_nccl()" "  "

     

       309
       309
       -
       

     

       323
       323
       +
           substituteInPlace pyproject.toml \

     

       324
       324
       +
             --replace-fail "setuptools>=62.3.0,<80.0" "setuptools"

     

       325
       325
       +
         ''

     

       326
       326
       +
         # Provide path to openssl binary for inductor code cache hash

     

       327
       327
       +
         # InductorError: FileNotFoundError: [Errno 2] No such file or directory: 'openssl'

     

       328
       328
       +
         + ''

     

       329
       329
       +
           substituteInPlace torch/_inductor/codecache.py \

     

       330
       330
       +
             --replace-fail '"openssl"' '"${lib.getExe openssl}"'

     

       331
       331
       +
         ''

     

       332
       332
       +
         + ''

     

       310
       333
        
           substituteInPlace cmake/public/cuda.cmake \

     

       311
       334
        
             --replace-fail \

     

       312
       335
        
               'message(FATAL_ERROR "Found two conflicting CUDA' \

     
···

       316
       339
        
               "# Upstream: set(CUDAToolkit_ROOT"

     

       317
       340
        
           substituteInPlace third_party/gloo/cmake/Cuda.cmake \

     

       318
       341
        
             --replace-warn "find_package(CUDAToolkit 7.0" "find_package(CUDAToolkit"

     

       319
       319
       -
       

     

       320
       320
       -
           # annotations (3.7), print_function (3.0), with_statement (2.6) are all supported

     

       342
       342
       +
         ''

     

       343
       343
       +
         # annotations (3.7), print_function (3.0), with_statement (2.6) are all supported

     

       344
       344
       +
         + ''

     

       321
       345
        
           sed -i -e "/from __future__ import/d" **.py

     

       322
       346
        
           substituteInPlace third_party/NNPACK/CMakeLists.txt \

     

       323
       347
        
             --replace-fail "PYTHONPATH=" 'PYTHONPATH=$ENV{PYTHONPATH}:'

     

       324
       324
       -
           # flag from cmakeFlags doesn't work, not clear why

     

       325
       325
       -
           # setting it at the top of NNPACK's own CMakeLists does

     

       348
       348
       +
         ''

     

       349
       349
       +
         # flag from cmakeFlags doesn't work, not clear why

     

       350
       350
       +
         # setting it at the top of NNPACK's own CMakeLists does

     

       351
       351
       +
         + ''

     

       326
       352
        
           sed -i '2s;^;set(PYTHON_SIX_SOURCE_DIR ${six.src})\n;' third_party/NNPACK/CMakeLists.txt

     

       327
       327
       -
       

     

       328
       328
       -
           # Ensure that torch profiler unwind uses addr2line from nix

     

       353
       353
       +
         ''

     

       354
       354
       +
         # Ensure that torch profiler unwind uses addr2line from nix

     

       355
       355
       +
         + ''

     

       329
       356
        
           substituteInPlace torch/csrc/profiler/unwind/unwind.cpp \

     

       330
       357
        
             --replace-fail 'addr2line_binary_ = "addr2line"' 'addr2line_binary_ = "${lib.getExe' binutils "addr2line"}"'

     

       331
       358
        
         ''

     
···

       333
       360
        
           # https://github.com/facebookincubator/gloo/pull/297

     

       334
       361
        
           substituteInPlace third_party/gloo/cmake/Hipify.cmake \

     

       335
       362
        
             --replace-fail "\''${HIPIFY_COMMAND}" "python \''${HIPIFY_COMMAND}"

     

       336
       336
       -
       

     

       337
       337
       -
           # Replace hard-coded rocm paths

     

       338
       338
       -
           substituteInPlace caffe2/CMakeLists.txt \

     

       339
       339
       -
             --replace-fail "hcc/include" "hip/include" \

     

       340
       340
       -
             --replace-fail "rocblas/include" "include/rocblas" \

     

       341
       341
       -
             --replace-fail "hipsparse/include" "include/hipsparse"

     

       342
       363
        
       

     

       343
       364
        
           # Doesn't pick up the environment variable?

     

       344
       365
        
           substituteInPlace third_party/kineto/libkineto/CMakeLists.txt \

     

       345
       366
        
             --replace-fail "\''$ENV{ROCM_SOURCE_DIR}" "${rocmtoolkit_joined}"

     

       346
       367
        
       

     

       347
       347
       -
           # Strangely, this is never set in cmake

     

       368
       368
       +
           # Workaround cmake error //include does not exist! in rocm-core-config.cmake

     

       369
       369
       +
           # Removing the call falls back to hip_version. Can likely be removed after ROCm 6.4 bump

     

       348
       370
        
           substituteInPlace cmake/public/LoadHIP.cmake \

     

       349
       349
       -
             --replace "set(ROCM_PATH \$ENV{ROCM_PATH})" \

     

       350
       350
       -
               "set(ROCM_PATH \$ENV{ROCM_PATH})''\nset(ROCM_VERSION ${lib.concatStrings (lib.intersperse "0" (lib.splitVersion rocmPackages.clr.version))})"

     

       371
       371
       +
             --replace-fail \

     

       372
       372
       +
               "find_package(rocm-core CONFIG)" \

     

       373
       373
       +
               ""

     

       351
       374
        
       

     

       352
       375
        
           # Use composable kernel as dependency, rather than built-in third-party

     

       353
       376
        
           substituteInPlace aten/src/ATen/CMakeLists.txt \

+26

pkgs/development/python-modules/torch/source/nvtx3-hpp-path-fix.patch

···

       1
       1
       +
       diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake

     

       2
       2
       +
       index a93386c27f8..7c6b98006bf 100644

     

       3
       3
       +
       --- a/cmake/Dependencies.cmake

     

       4
       4
       +
       +++ b/cmake/Dependencies.cmake

     

       5
       5
       +
       @@ -967,7 +967,7 @@ endif()

     

       6
       6
       +
       

     

       7
       7
       +
        # ---[ nvtx

     

       8
       8
       +
        if(USE_SYSTEM_NVTX)

     

       9
       9
       +
       -  find_path(nvtx3_dir NAMES nvtx3 PATHS ${CUDA_INCLUDE_DIRS})

     

       10
       10
       +
       +  find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS ${CUDA_INCLUDE_DIRS})

     

       11
       11
       +
          find_package_handle_standard_args(nvtx3 DEFAULT_MSG nvtx3_dir)

     

       12
       12
       +
          if(NOT nvtx3_FOUND)

     

       13
       13
       +
            message(WARNING "Cannot find system NVTX3, find shipped NVTX3 instead")

     

       14
       14
       +
       @@ -977,9 +977,10 @@ if(NOT TARGET CUDA::nvtx3)

     

       15
       15
       +
          add_library(CUDA::nvtx3 INTERFACE IMPORTED)

     

       16
       16
       +
        endif()

     

       17
       17
       +
        if(NOT nvtx3_dir)

     

       18
       18
       +
       -  find_path(nvtx3_dir NAMES nvtx3 PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)

     

       19
       19
       +
       -  target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}")

     

       20
       20
       +
       +  find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)

     

       21
       21
       +
        endif()

     

       22
       22
       +
       +target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}")

     

       23
       23
       +
       +message(STATUS "Using NVTX3 include directory: ${nvtx3_dir}")

     

       24
       24
       +
       

     

       25
       25
       +
       

     

       26
       26
       +
        # ---[ HIP

+84 -83

pkgs/development/python-modules/torch/source/src.nix

···

       4
       4
        
         fetchFromGitHub,

     

       5
       5
        
         runCommand,

     

       6
       6
        
       }:

     

       7
       7
       -
       assert version == "2.7.1";

     

       7
       7
       +
       assert version == "2.8.0";

     

       8
       8
        
       (rec {

     

       9
       9
        
         src_asmjit = fetchFromGitHub {

     

       10
       10
        
           owner = "asmjit";

     

       11
       11
        
           repo = "asmjit";

     

       12
       12
       -
           rev = "d3fbf7c9bc7c1d1365a94a45614b91c5a3706b81";

     

       13
       13
       -
           hash = "sha256-0Wv9dxrh9GfajTFb+NpguqqSWH0mqJAj03bxFVJbft8=";

     

       12
       12
       +
           rev = "e5d7c0bd5d9aec44d68830187138149e6a8c4e32";

     

       13
       13
       +
           hash = "sha256-sI0/9szBMvopQAmVcZSU4D/oaZYdb08AHDSZKy/Qz1g=";

     

       14
       14
        
         };

     

       15
       15
        
         src_benchmark = fetchFromGitHub {

     

       16
       16
        
           owner = "google";

     

       17
       17
        
           repo = "benchmark";

     

       18
       18
       -
           rev = "0d98dba29d66e93259db7daa53a9327df767a415";

     

       19
       19
       -
           hash = "sha256-yUiFxi80FWBmTZgqmqTMf9oqcBeg3o4I4vKd4djyRWY=";

     

       18
       18
       +
           rev = "299e5928955cc62af9968370293b916f5130916f";

     

       19
       19
       +
           hash = "sha256-iPK3qLrZL2L08XW1a7SGl7GAt5InQ5nY+Dn8hBuxSOg=";

     

       20
       20
        
         };

     

       21
       21
        
         src_benchmark_opentelemetry-cpp = fetchFromGitHub {

     

       22
       22
        
           owner = "google";

     
···

       48
       48
        
           rev = "8086bbe3a78d931eb96fe12fdc014082e18d18d3";

     

       49
       49
        
           hash = "sha256-fyL1SzRs5CXW5CWy6kCN1y1xX6cG+ur7iQlbKX2zbCM=";

     

       50
       50
        
         };

     

       51
       51
       +
         src_composable_kernel_fbgemm = fetchFromGitHub {

     

       52
       52
       +
           owner = "jwfromm";

     

       53
       53
       +
           repo = "composable_kernel";

     

       54
       54
       +
           rev = "4a61bdd4bd4ed730e078aebc7c0fcf046ff29406";

     

       55
       55
       +
           hash = "sha256-CxcpvW4QxkUDB2zMz7NB6Rt9jXjJeGVExfrYbn9ef5I=";

     

       56
       56
       +
         };

     

       51
       57
        
         src_composable_kernel_flash-attention = fetchFromGitHub {

     

       52
       58
        
           owner = "ROCm";

     

       53
       59
        
           repo = "composable_kernel";

     
···

       57
       63
        
         src_cpp-httplib = fetchFromGitHub {

     

       58
       64
        
           owner = "yhirose";

     

       59
       65
        
           repo = "cpp-httplib";

     

       60
       60
       -
           rev = "3b6597bba913d51161383657829b7e644e59c006";

     

       61
       61
       -
           hash = "sha256-dd9NckF1mGhQOyV1LO07QyP51l1kSpYQOH0GkG4v2eE=";

     

       66
       66
       +
           rev = "3af7f2c16147f3fbc6e4d717032daf505dc1652c";

     

       67
       67
       +
           hash = "sha256-t/ddZjKelnXQdXQvZgv9pQcJt3M2rwgbtTQNW/T6Gpk=";

     

       62
       68
        
         };

     

       63
       69
        
         src_cpr = fetchFromGitHub {

     

       64
       70
        
           owner = "libcpr";

     
···

       69
       75
        
         src_cpuinfo = fetchFromGitHub {

     

       70
       76
        
           owner = "pytorch";

     

       71
       77
        
           repo = "cpuinfo";

     

       72
       72
       -
           rev = "1e83a2fdd3102f65c6f1fb602c1b320486218a99";

     

       73
       73
       -
           hash = "sha256-28cFACca+NYE8oKlP5aWXNCLeEjhWqJ6gRnFI+VxDvg=";

     

       78
       78
       +
           rev = "5e3d2445e6a84d9599bee2bf78edbb4d80865e1d";

     

       79
       79
       +
           hash = "sha256-muclbtayPbbxmjTBKJlvTk2Apsx8YwhTu76nKMlSnL0=";

     

       74
       80
        
         };

     

       75
       81
        
         src_cpuinfo_fbgemm = fetchFromGitHub {

     

       76
       82
        
           owner = "pytorch";

     

       77
       83
        
           repo = "cpuinfo";

     

       78
       78
       -
           rev = "ed8b86a253800bafdb7b25c5c399f91bff9cb1f3";

     

       79
       79
       -
           hash = "sha256-YRqBU83AjxbSE5zquhi4iIiJna/qFWA0jo2GBifqzi8=";

     

       84
       84
       +
           rev = "6543fec09b2f04ac4a666882998b534afc9c1349";

     

       85
       85
       +
           hash = "sha256-M33853TIMNTJwBjxIG8qeTPwXlX8b8uKiCrdnlAtAA8=";

     

       80
       86
        
         };

     

       81
       87
        
         src_cudnn-frontend = fetchFromGitHub {

     

       82
       88
        
           owner = "NVIDIA";

     

       83
       89
        
           repo = "cudnn-frontend";

     

       84
       84
       -
           rev = "91b7532f3386768bba4f444ee7672b497f34da8a";

     

       85
       85
       -
           hash = "sha256-Ks07ApADA3xQQ+N5BIfvDG+djCvxYL9btF8Aw7TuMvQ=";

     

       90
       90
       +
           rev = "666996fe3960f27170d1527e5579ba24c8d3380a";

     

       91
       91
       +
           hash = "sha256-/ae5dNjqkn5dGciGSO+Pn9pzJg+hHWqqQCPcqggWezo=";

     

       86
       92
        
         };

     

       87
       93
        
         src_cutlass = fetchFromGitHub {

     

       88
       94
        
           owner = "NVIDIA";

     

       89
       95
        
           repo = "cutlass";

     

       90
       90
       -
           rev = "afa1772203677c5118fcd82537a9c8fefbcc7008";

     

       91
       91
       -
           hash = "sha256-oIzlbKRdOh6gp6nRZ8udLSqleBFoFtgM7liCBlHZLOk=";

     

       96
       96
       +
           rev = "ad7b2f5e84fcfa124cb02b91d5bd26d238c0459e";

     

       97
       97
       +
           hash = "sha256-teziPNA9csYvhkG5t2ht8W8x5+1YGGbHm8VKx4JoxgI=";

     

       92
       98
        
         };

     

       93
       99
        
         src_cutlass_fbgemm = fetchFromGitHub {

     

       94
       94
       -
           owner = "NVIDIA";

     

       100
       100
       +
           owner = "jwfromm";

     

       95
       101
        
           repo = "cutlass";

     

       96
       96
       -
           rev = "fc9ebc645b63f3a6bc80aaefde5c063fb72110d6";

     

       97
       97
       -
           hash = "sha256-e2SwXNNwjl/1fV64b+mOJvwGDYeO1LFcqZGbNten37U=";

     

       102
       102
       +
           rev = "3ed8d2ec4ba35ef5d9d8353826209b6f868f63d3";

     

       103
       103
       +
           hash = "sha256-NntohGvqs6fbWusi2Qv5uzCJhMAfBv8qYoFi38D+mzk=";

     

       98
       104
        
         };

     

       99
       105
        
         src_cutlass_flash-attention = fetchFromGitHub {

     

       100
       106
        
           owner = "NVIDIA";

     
···

       114
       120
        
           rev = "7d04a0053a845370ae06ce317a22a48e9edcc74e";

     

       115
       121
        
           hash = "sha256-Je6wAz+uJ/AiAnSZVQ4+pGajZ8DymS0qI9ekB8fGYOo=";

     

       116
       122
        
         };

     

       117
       117
       -
         src_eigen = fetchFromGitLab {

     

       118
       118
       -
           domain = "gitlab.com";

     

       119
       119
       -
           owner = "libeigen";

     

       120
       120
       -
           repo = "eigen";

     

       121
       121
       -
           rev = "3147391d946bb4b6c68edd901f2add6ac1f31f8c";

     

       122
       122
       -
           hash = "sha256-1/4xMetKMDOgZgzz3WMxfHUEpmdAm52RqZvz6i0mLEw=";

     

       123
       123
       -
         };

     

       124
       123
        
         src_fbgemm = fetchFromGitHub {

     

       125
       124
        
           owner = "pytorch";

     

       126
       125
        
           repo = "fbgemm";

     

       127
       127
       -
           rev = "dbc3157bf256f1339b3fa1fef2be89ac4078be0e";

     

       128
       128
       -
           hash = "sha256-PJiFtLnPA6IgxZ2sXIcyyjFRGtb+sG5y2hiWEwFuBOU=";

     

       126
       126
       +
           rev = "157e88b750c452bef2ab4653fe9d1eeb151ce4c3";

     

       127
       127
       +
           hash = "sha256-Ka8/4gBsbtKNhKM/cWg1NmlKjVeBZvS+yS9SQQxb34A=";

     

       129
       128
        
         };

     

       130
       129
        
         src_fbjni = fetchFromGitHub {

     

       131
       130
        
           owner = "facebookincubator";

     
···

       142
       141
        
         src_flatbuffers = fetchFromGitHub {

     

       143
       142
        
           owner = "google";

     

       144
       143
        
           repo = "flatbuffers";

     

       145
       145
       -
           rev = "01834de25e4bf3975a9a00e816292b1ad0fe184b";

     

       146
       146
       -
           hash = "sha256-h0lF7jf1cDVVyqhUCi7D0NoZ3b4X/vWXsFplND80lGs=";

     

       144
       144
       +
           rev = "a2cd1ea3b6d3fee220106b5fed3f7ce8da9eb757";

     

       145
       145
       +
           hash = "sha256-6L6Eb+2xGXEqLYITWsNNPW4FTvfPFSmChK4hLusk5gU=";

     

       147
       146
        
         };

     

       148
       147
        
         src_fmt = fetchFromGitHub {

     

       149
       148
        
           owner = "fmtlib";

     

       150
       149
        
           repo = "fmt";

     

       151
       151
       -
           rev = "123913715afeb8a437e6388b4473fcc4753e1c9a";

     

       152
       152
       -
           hash = "sha256-sUbxlYi/Aupaox3JjWFqXIjcaQa0LFjclQAOleT+FRA=";

     

       150
       150
       +
           rev = "40626af88bd7df9a5fb80be7b25ac85b122d6c21";

     

       151
       151
       +
           hash = "sha256-sAlU5L/olxQUYcv8euVYWTTB8TrVeQgXLHtXy8IMEnU=";

     

       153
       152
        
         };

     

       154
       153
        
         src_fmt_dynolog = fetchFromGitHub {

     

       155
       154
        
           owner = "fmtlib";

     
···

       200
       199
        
           hash = "sha256-xqRp9vaauBkKz2CXbh/Z4TWqhaUtqfbsSlbYZR/kW9s=";

     

       201
       200
        
         };

     

       202
       201
        
         src_gloo = fetchFromGitHub {

     

       203
       203
       -
           owner = "facebookincubator";

     

       202
       202
       +
           owner = "pytorch";

     

       204
       203
        
           repo = "gloo";

     

       205
       205
       -
           rev = "5354032ea08eadd7fc4456477f7f7c6308818509";

     

       206
       206
       -
           hash = "sha256-JMLtxyQz7jechJ5DmMq0guOfL9leI6khdI9g/5Ckgfc=";

     

       204
       204
       +
           rev = "c7b7b022c124d9643957d9bd55f57ac59fce8fa2";

     

       205
       205
       +
           hash = "sha256-pZ08gs6wQTZNVDX9uuaQZvw5JKCps8EALegNF0UVV3c=";

     

       207
       206
        
         };

     

       208
       207
        
         src_googletest = fetchFromGitHub {

     

       209
       208
        
           owner = "google";

     

       210
       209
        
           repo = "googletest";

     

       211
       211
       -
           rev = "b514bdc898e2951020cbdca1304b75f5950d1f59";

     

       212
       212
       -
           hash = "sha256-1OJ2SeSscRBNr7zZ/a8bJGIqAnhkg45re0j3DtPfcXM=";

     

       210
       210
       +
           rev = "52eb8108c5bdec04579160ae17225d66034bd723";

     

       211
       211
       +
           hash = "sha256-HIHMxAUR4bjmFLoltJeIAVSulVQ6kVuIT2Ku+lwAx/4=";

     

       213
       212
        
         };

     

       214
       213
        
         src_googletest_dynolog = fetchFromGitHub {

     

       215
       214
        
           owner = "google";

     
···

       220
       219
        
         src_googletest_fbgemm = fetchFromGitHub {

     

       221
       220
        
           owner = "google";

     

       222
       221
        
           repo = "googletest";

     

       223
       223
       -
           rev = "cbf019de22c8dd37b2108da35b2748fd702d1796";

     

       224
       224
       -
           hash = "sha256-G6NihPly7czG2NOX66kFfcf5ya+XRrUWt4SP1Y9JPzs=";

     

       222
       222
       +
           rev = "f8d7d77c06936315286eb55f8de22cd23c188571";

     

       223
       223
       +
           hash = "sha256-t0RchAHTJbuI5YW4uyBPykTvcjy90JW9AOPNjIhwh6U=";

     

       225
       224
        
         };

     

       226
       225
        
         src_googletest_kineto = fetchFromGitHub {

     

       227
       226
        
           owner = "google";

     
···

       262
       261
        
         src_hipify_torch = fetchFromGitHub {

     

       263
       262
        
           owner = "ROCmSoftwarePlatform";

     

       264
       263
        
           repo = "hipify_torch";

     

       265
       265
       -
           rev = "23f53b025b466d8ec3c45d52290d3442f7fbe6b1";

     

       266
       266
       -
           hash = "sha256-ohbGKy0sxa5pQy9EwsZk2UWmjveCZaJu/PEK2MLbjII=";

     

       264
       264
       +
           rev = "a4337c69fe0e2552a7b7b0669178926beeed828c";

     

       265
       265
       +
           hash = "sha256-B0+tDjSlZ9C5IAAgteRIgwaJNnptpp1jOP3hTF5AdOw=";

     

       267
       266
        
         };

     

       268
       267
        
         src_ideep = fetchFromGitHub {

     

       269
       268
        
           owner = "intel";

     
···

       274
       273
        
         src_ittapi = fetchFromGitHub {

     

       275
       274
        
           owner = "intel";

     

       276
       275
        
           repo = "ittapi";

     

       277
       277
       -
           rev = "5b8a7d7422611c3a0d799fb5fc5dd4abfae35b42";

     

       278
       278
       -
           hash = "sha256-VxJky2TF3RcIMqjNaAK/mvpC0afkwpAsY0cD6Ergkls=";

     

       276
       276
       +
           rev = "dec1d23ca65ab069d225dfe40dea14f455170959";

     

       277
       277
       +
           hash = "sha256-aP95qtYi1s8go6HbXW6hDd0XzdWQhOwXLJBNPCIkSCI=";

     

       279
       278
        
         };

     

       280
       279
        
         src_json = fetchFromGitHub {

     

       281
       280
        
           owner = "nlohmann";

     

       282
       281
        
           repo = "json";

     

       283
       283
       -
           rev = "87cda1d6646592ac5866dc703c8e1839046a6806";

     

       284
       284
       -
           hash = "sha256-lXYJGWwLyQPqvxnDRWoDLXdjiD81r1eNHi7vRdbIuJ0=";

     

       282
       282
       +
           rev = "55f93686c01528224f448c19128836e7df245f72";

     

       283
       283
       +
           hash = "sha256-cECvDOLxgX7Q9R3IE86Hj9JJUxraDQvhoyPDF03B2CY=";

     

       285
       284
        
         };

     

       286
       285
        
         src_json_dynolog = fetchFromGitHub {

     

       287
       286
        
           owner = "nlohmann";

     
···

       289
       288
        
           rev = "4f8fba14066156b73f1189a2b8bd568bde5284c5";

     

       290
       289
        
           hash = "sha256-DTsZrdB9GcaNkx7ZKxcgCA3A9ShM5icSF0xyGguJNbk=";

     

       291
       290
        
         };

     

       291
       291
       +
         src_json_fbgemm = fetchFromGitHub {

     

       292
       292
       +
           owner = "nlohmann";

     

       293
       293
       +
           repo = "json";

     

       294
       294
       +
           rev = "9cca280a4d0ccf0c08f47a99aa71d1b0e52f8d03";

     

       295
       295
       +
           hash = "sha256-7F0Jon+1oWL7uqet5i1IgHX0fUw/+z0QwEcA3zs5xHg=";

     

       296
       296
       +
         };

     

       292
       297
        
         src_json_opentelemetry-cpp = fetchFromGitHub {

     

       293
       298
        
           owner = "nlohmann";

     

       294
       299
        
           repo = "json";

     
···

       298
       303
        
         src_kineto = fetchFromGitHub {

     

       299
       304
        
           owner = "pytorch";

     

       300
       305
        
           repo = "kineto";

     

       301
       301
       -
           rev = "a054a4be0db117c579a21747debf19c863631f26";

     

       302
       302
       -
           hash = "sha256-FyxMiDoDrjI8JZZNxK4S18yjULkKLUvqLQsCyxaTLSg=";

     

       306
       306
       +
           rev = "5e7501833f1021ce6f618572d3baf657b6319658";

     

       307
       307
       +
           hash = "sha256-MQNGrWKU1l/lXo4G2hDFaQeNtyQj+pB5sTqm3piPYrY=";

     

       303
       308
        
         };

     

       304
       309
        
         src_kleidiai = fetchFromGitHub {

     

       305
       310
        
           owner = "ARM-software";

     

       306
       311
        
           repo = "kleidiai";

     

       307
       307
       -
           rev = "ef685a13cfbe8d418aa2ed34350e21e4938358b6";

     

       308
       308
       -
           hash = "sha256-5RS2o+163/6Q+0y0FuWz6OBXw3FKxZINDkGOqlmd2N0=";

     

       312
       312
       +
           rev = "cca02c2f69dd18e1f12647c1c0bdc8cf90e680c7";

     

       313
       313
       +
           hash = "sha256-+75YvMEJN57cz/jzbqC3E0O7BalfaIfzswRZdpOMx0Q=";

     

       309
       314
        
         };

     

       310
       315
        
         src_libnop = fetchFromGitHub {

     

       311
       316
        
           owner = "google";

     
···

       322
       327
        
         src_mimalloc = fetchFromGitHub {

     

       323
       328
        
           owner = "microsoft";

     

       324
       329
        
           repo = "mimalloc";

     

       325
       325
       -
           rev = "b66e3214d8a104669c2ec05ae91ebc26a8f5ab78";

     

       326
       326
       -
           hash = "sha256-uwuqln08Hx1d2l7GNn8/8hzOA1Pmzob5g17XgFb+blg=";

     

       330
       330
       +
           rev = "94036de6fe20bfd8a73d4a6d142fcf532ea604d9";

     

       331
       331
       +
           hash = "sha256-B0gngv16WFLBtrtG5NqA2m5e95bYVcQraeITcOX9A74=";

     

       327
       332
        
         };

     

       328
       333
        
         src_mkl-dnn = fetchFromGitHub {

     

       329
       334
        
           owner = "intel";

     
···

       340
       345
        
         src_NVTX = fetchFromGitHub {

     

       341
       346
        
           owner = "NVIDIA";

     

       342
       347
        
           repo = "NVTX";

     

       343
       343
       -
           rev = "e170594ac7cf1dac584da473d4ca9301087090c1";

     

       344
       344
       -
           hash = "sha256-n34BPxRnAW301ba1lXqSlGh7jaPqNjpp45GnJ+yDapI=";

     

       348
       348
       +
           rev = "2942f167cc30c5e3a44a2aecd5b0d9c07ff61a07";

     

       349
       349
       +
           hash = "sha256-MXluy/I5+SaRx2aF64qF4XZ+u67ERAB9TftbOvYt4GE=";

     

       345
       350
        
         };

     

       346
       351
        
         src_onnx = fetchFromGitHub {

     

       347
       352
        
           owner = "onnx";

     

       348
       353
        
           repo = "onnx";

     

       349
       349
       -
           rev = "b8baa8446686496da4cc8fda09f2b6fe65c2a02c";

     

       350
       350
       -
           hash = "sha256-9oORW0YlQ6SphqfbjcYb0dTlHc+1gzy9quH/Lj6By8Q=";

     

       354
       354
       +
           rev = "e709452ef2bbc1d113faf678c24e6d3467696e83";

     

       355
       355
       +
           hash = "sha256-UhtF+CWuyv5/Pq/5agLL4Y95YNP63W2BraprhRqJOag=";

     

       351
       356
        
         };

     

       352
       357
        
         src_opentelemetry-cpp = fetchFromGitHub {

     

       353
       358
        
           owner = "open-telemetry";

     
···

       382
       387
        
         src_pocketfft = fetchFromGitHub {

     

       383
       388
        
           owner = "mreineck";

     

       384
       389
        
           repo = "pocketfft";

     

       385
       385
       -
           rev = "9d3ab05a7fffbc71a492bc6a17be034e83e8f0fe";

     

       386
       386
       -
           hash = "sha256-RSbimayr8Np7YP0aUo1MNusFmhi9jjDfgGXbiISR+/8=";

     

       390
       390
       +
           rev = "0fa0ef591e38c2758e3184c6c23e497b9f732ffa";

     

       391
       391
       +
           hash = "sha256-Fu786IHiU6Bl66gZ/UJmqOROjlya3viLyzOxwdZVi9c=";

     

       387
       392
        
         };

     

       388
       393
        
         src_prometheus-cpp = fetchFromGitHub {

     

       389
       394
        
           owner = "jupp0r";

     
···

       415
       420
        
           rev = "a2e59f0e7065404b44dfe92a28aca47ba1378dc4";

     

       416
       421
        
           hash = "sha256-SNLdtrOjaC3lGHN9MAqTf51U9EzNKQLyTMNPe0GcdrU=";

     

       417
       422
        
         };

     

       418
       418
       -
         src_pybind11_onnx = fetchFromGitHub {

     

       419
       419
       -
           owner = "pybind";

     

       420
       420
       -
           repo = "pybind11";

     

       421
       421
       -
           rev = "3e9dfa2866941655c56877882565e7577de6fc7b";

     

       422
       422
       -
           hash = "sha256-DVkI5NxM5uME9m3PFYVpJOOa2j+yjL6AJn76fCTv2nE=";

     

       423
       423
       -
         };

     

       424
       423
        
         src_pybind11_tensorpipe = fetchFromGitHub {

     

       425
       424
        
           owner = "pybind";

     

       426
       425
        
           repo = "pybind11";

     
···

       430
       429
        
         src_pytorch = fetchFromGitHub {

     

       431
       430
        
           owner = "pytorch";

     

       432
       431
        
           repo = "pytorch";

     

       433
       433
       -
           rev = "v2.7.1";

     

       434
       434
       -
           hash = "sha256-p/SkVM6N4XGr44WB9ZaMLu6nfJGcaKHfb1hDY6qbZBw=";

     

       432
       432
       +
           rev = "v2.8.0";

     

       433
       433
       +
           hash = "sha256-tFEpcgj0HiJcyBiZMtIrBrnmiCJApfTC1BgOXEGvqCo=";

     

       435
       434
        
         };

     

       436
       435
        
         src_sleef = fetchFromGitHub {

     

       437
       436
        
           owner = "shibatch";

     

       438
       437
        
           repo = "sleef";

     

       439
       439
       -
           rev = "56e1f79cb140fb9326d612d0be06b5250565cade";

     

       440
       440
       -
           hash = "sha256-5hha7c/Lu6fkbXGlDieoJP6n2bnjY5iPp+hm0f//0ek=";

     

       438
       438
       +
           rev = "5a1d179df9cf652951b59010a2d2075372d67f68";

     

       439
       439
       +
           hash = "sha256-bjT+F7/nyiB4f0T06/flbpIWFZbUxjf1TjWMe3112Ig=";

     

       441
       440
        
         };

     

       442
       441
        
         src_tensorpipe = fetchFromGitHub {

     

       443
       442
        
           owner = "pytorch";

     
···

       454
       453
        
         src_VulkanMemoryAllocator = fetchFromGitHub {

     

       455
       454
        
           owner = "GPUOpen-LibrariesAndSDKs";

     

       456
       455
        
           repo = "VulkanMemoryAllocator";

     

       457
       457
       -
           rev = "a6bfc237255a6bac1513f7c1ebde6d8aed6b5191";

     

       458
       458
       -
           hash = "sha256-urUebQaPTgCECmm4Espri1HqYGy0ueAqTBu/VSiX/8I=";

     

       456
       456
       +
           rev = "1d8f600fd424278486eade7ed3e877c99f0846b1";

     

       457
       457
       +
           hash = "sha256-TPEqV8uHbnyphLG0A+b2tgLDQ6K7a2dOuDHlaFPzTeE=";

     

       459
       458
        
         };

     

       460
       459
        
         src_XNNPACK = fetchFromGitHub {

     

       461
       460
        
           owner = "google";

     
···

       470
       469
        
         src_civetweb_recursive = src_civetweb;

     

       471
       470
        
         src_clang-cindex-python3_recursive = src_clang-cindex-python3;

     

       472
       471
        
         src_composable_kernel_recursive = src_composable_kernel;

     

       472
       472
       +
         src_composable_kernel_fbgemm_recursive = src_composable_kernel_fbgemm;

     

       473
       473
        
         src_composable_kernel_flash-attention_recursive = src_composable_kernel_flash-attention;

     

       474
       474
        
         src_cpp-httplib_recursive = src_cpp-httplib;

     

       475
       475
        
         src_cpr_recursive = src_cpr;

     
···

       499
       499
        
           chmod u+w $out/third_party/pfs

     

       500
       500
        
           cp -r ${src_pfs_recursive}/* $out/third_party/pfs

     

       501
       501
        
         '';

     

       502
       502
       -
         src_eigen_recursive = src_eigen;

     

       503
       502
        
         src_fbgemm_recursive = runCommand "fbgemm" { } ''

     

       504
       503
        
           cp -r ${src_fbgemm} $out

     

       505
       505
       -
           chmod u+w $out/third_party/asmjit

     

       506
       506
       -
           cp -r ${src_asmjit_recursive}/* $out/third_party/asmjit

     

       507
       507
       -
           chmod u+w $out/third_party/cpuinfo

     

       508
       508
       -
           cp -r ${src_cpuinfo_fbgemm_recursive}/* $out/third_party/cpuinfo

     

       509
       509
       -
           chmod u+w $out/third_party/cutlass

     

       510
       510
       -
           cp -r ${src_cutlass_fbgemm_recursive}/* $out/third_party/cutlass

     

       511
       511
       -
           chmod u+w $out/third_party/googletest

     

       512
       512
       -
           cp -r ${src_googletest_fbgemm_recursive}/* $out/third_party/googletest

     

       513
       513
       -
           chmod u+w $out/third_party/hipify_torch

     

       514
       514
       -
           cp -r ${src_hipify_torch_recursive}/* $out/third_party/hipify_torch

     

       504
       504
       +
           chmod u+w $out/external/asmjit

     

       505
       505
       +
           cp -r ${src_asmjit_recursive}/* $out/external/asmjit

     

       506
       506
       +
           chmod u+w $out/external/composable_kernel

     

       507
       507
       +
           cp -r ${src_composable_kernel_fbgemm_recursive}/* $out/external/composable_kernel

     

       508
       508
       +
           chmod u+w $out/external/cpuinfo

     

       509
       509
       +
           cp -r ${src_cpuinfo_fbgemm_recursive}/* $out/external/cpuinfo

     

       510
       510
       +
           chmod u+w $out/external/cutlass

     

       511
       511
       +
           cp -r ${src_cutlass_fbgemm_recursive}/* $out/external/cutlass

     

       512
       512
       +
           chmod u+w $out/external/googletest

     

       513
       513
       +
           cp -r ${src_googletest_fbgemm_recursive}/* $out/external/googletest

     

       514
       514
       +
           chmod u+w $out/external/hipify_torch

     

       515
       515
       +
           cp -r ${src_hipify_torch_recursive}/* $out/external/hipify_torch

     

       516
       516
       +
           chmod u+w $out/external/json

     

       517
       517
       +
           cp -r ${src_json_fbgemm_recursive}/* $out/external/json

     

       515
       518
        
         '';

     

       516
       519
        
         src_fbjni_recursive = src_fbjni;

     

       517
       520
        
         src_flash-attention_recursive = runCommand "flash-attention" { } ''

     
···

       554
       557
        
         src_ittapi_recursive = src_ittapi;

     

       555
       558
        
         src_json_recursive = src_json;

     

       556
       559
        
         src_json_dynolog_recursive = src_json_dynolog;

     

       560
       560
       +
         src_json_fbgemm_recursive = src_json_fbgemm;

     

       557
       561
        
         src_json_opentelemetry-cpp_recursive = src_json_opentelemetry-cpp;

     

       558
       562
        
         src_kineto_recursive = runCommand "kineto" { } ''

     

       559
       563
        
           cp -r ${src_kineto} $out

     
···

       574
       578
        
         src_onnx_recursive = runCommand "onnx" { } ''

     

       575
       579
        
           cp -r ${src_onnx} $out

     

       576
       580
        
           chmod u+w $out/third_party/pybind11

     

       577
       577
       -
           cp -r ${src_pybind11_onnx_recursive}/* $out/third_party/pybind11

     

       581
       581
       +
           cp -r ${src_pybind11_recursive}/* $out/third_party/pybind11

     

       578
       582
        
         '';

     

       579
       583
        
         src_opentelemetry-cpp_recursive = runCommand "opentelemetry-cpp" { } ''

     

       580
       584
        
           cp -r ${src_opentelemetry-cpp} $out

     
···

       617
       621
        
         src_psimd_recursive = src_psimd;

     

       618
       622
        
         src_pthreadpool_recursive = src_pthreadpool;

     

       619
       623
        
         src_pybind11_recursive = src_pybind11;

     

       620
       620
       -
         src_pybind11_onnx_recursive = src_pybind11_onnx;

     

       621
       624
        
         src_pybind11_tensorpipe_recursive = runCommand "pybind11_tensorpipe" { } ''

     

       622
       625
        
           cp -r ${src_pybind11_tensorpipe} $out

     

       623
       626
        
           chmod u+w $out/tools/clang

     
···

       639
       642
        
           cp -r ${src_cudnn-frontend_recursive}/* $out/third_party/cudnn_frontend

     

       640
       643
        
           chmod u+w $out/third_party/cutlass

     

       641
       644
        
           cp -r ${src_cutlass_recursive}/* $out/third_party/cutlass

     

       642
       642
       -
           chmod u+w $out/third_party/eigen

     

       643
       643
       -
           cp -r ${src_eigen_recursive}/* $out/third_party/eigen

     

       644
       645
        
           chmod u+w $out/third_party/fbgemm

     

       645
       646
        
           cp -r ${src_fbgemm_recursive}/* $out/third_party/fbgemm

     

       646
       647
        
           chmod u+w $out/third_party/flash-attention