commit 1577208185498388dd3c54d370d8b789f6275f88 · pyrox.dev/nixpkgs

+6 -5

pkgs/development/compilers/hip/default.nix

···

       14
       14
        
       , numactl

     

       15
       15
        
       , perl

     

       16
       16
        
       , python3

     

       17
       17
       +
       , python3Packages

     

       17
       18
        
       , rocclr

     

       18
       19
        
       , rocm-comgr

     

       19
       20
        
       , rocm-device-libs

     
···

       29
       30
        
       let

     

       30
       31
        
         hip = stdenv.mkDerivation rec {

     

       31
       32
        
           pname = "hip";

     

       32
       32
       -
           version = "5.3.0";

     

       33
       33
       +
           version = "5.3.1";

     

       33
       34
        
       

     

       34
       35
        
           src = fetchFromGitHub {

     

       35
       36
        
             owner = "ROCm-Developer-Tools";

     

       36
       37
        
             repo = "HIP";

     

       37
       38
        
             rev = "rocm-${version}";

     

       38
       38
       -
             hash = "sha256-UAodlVUiTU4n/EyvTIuQekTGh4izmBjKCRXOHXVKY4M=";

     

       39
       39
       +
             hash = "sha256-kmRvrwnT0h2dBMI+H9d1vmeW3TmDBD+qW4YYhaMV2dE=";

     

       39
       40
        
           };

     

       40
       41
        
       

     

       41
       42
        
           patches = [

     
···

       109
       110
        
       in

     

       110
       111
        
       stdenv.mkDerivation rec {

     

       111
       112
        
         pname = "hip";

     

       112
       112
       -
         version = "5.3.0";

     

       113
       113
       +
         version = "5.3.1";

     

       113
       114
        
       

     

       114
       115
        
         src = fetchFromGitHub {

     

       115
       116
        
           owner = "ROCm-Developer-Tools";

     

       116
       117
        
           repo = "hipamd";

     

       117
       118
        
           rev = "rocm-${version}";

     

       118
       118
       -
           hash = "sha256-gZGZiDP/HbdmzLQkG9Jq9lyMP9hoD6UzTMiX9cUmQNA=";

     

       119
       119
       +
           hash = "sha256-i7hT/j+V0LT6Va2XcQyyKXF1guoIyhcOHvn842wCRx4=";

     

       119
       120
        
         };

     

       120
       121
        
       

     

       121
       122
        
         nativeBuildInputs = [ cmake python3 makeWrapper perl ];

     

       122
       122
       -
         buildInputs = [ libxml2 numactl libglvnd libX11 ];

     

       123
       123
       +
         buildInputs = [ libxml2 numactl libglvnd libX11 python3Packages.cppheaderparser ];

     

       123
       124
        
         propagatedBuildInputs = [

     

       124
       125
        
           clang

     

       125
       126
        
           llvm

+2 -2

pkgs/development/compilers/llvm/rocm/default.nix

···

       1
       1
        
       { stdenv, lib, buildPackages, fetchFromGitHub, callPackage, wrapCCWith, overrideCC }:

     

       2
       2
        
       

     

       3
       3
        
       let

     

       4
       4
       -
         version = "5.3.0";

     

       4
       4
       +
         version = "5.3.1";

     

       5
       5
        
         src = fetchFromGitHub {

     

       6
       6
        
           owner = "RadeonOpenCompute";

     

       7
       7
        
           repo = "llvm-project";

     

       8
       8
        
           rev = "rocm-${version}";

     

       9
       9
       -
           hash = "sha256-MN7W4Gl6+a1nCozdn9gMzIXOiBPquoOP87x26boeSCA=";

     

       9
       9
       +
           hash = "sha256-IKo7N8wWvh5PBrZ2mh1Vu5s3uUXhanqYtC4qLV/+JBs=";

     

       10
       10
        
         };

     

       11
       11
        
       in rec {

     

       12
       12
        
         clang = wrapCCWith rec {

+44

pkgs/development/libraries/clang-ocl/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-device-libs

     

       7
       7
       +
       , clang

     

       8
       8
       +
       }:

     

       9
       9
       +
       

     

       10
       10
       +
       stdenv.mkDerivation rec {

     

       11
       11
       +
         pname = "clang-ocl";

     

       12
       12
       +
         rocmVersion = "5.3.1";

     

       13
       13
       +
         version = rocmVersion;

     

       14
       14
       +
       

     

       15
       15
       +
         src = fetchFromGitHub {

     

       16
       16
       +
           owner = "RadeonOpenCompute";

     

       17
       17
       +
           repo = "clang-ocl";

     

       18
       18
       +
           rev = "rocm-${rocmVersion}";

     

       19
       19
       +
           hash = "sha256-uMSvcVJj+me2E+7FsXZ4l4hTcK6uKEegXpkHGcuist0=";

     

       20
       20
       +
         };

     

       21
       21
       +
       

     

       22
       22
       +
         nativeBuildInputs = [

     

       23
       23
       +
           cmake

     

       24
       24
       +
           rocm-cmake

     

       25
       25
       +
           clang

     

       26
       26
       +
         ];

     

       27
       27
       +
       

     

       28
       28
       +
         buildInputs = [

     

       29
       29
       +
           rocm-device-libs

     

       30
       30
       +
         ];

     

       31
       31
       +
       

     

       32
       32
       +
         cmakeFlags = [

     

       33
       33
       +
           "-DCMAKE_C_COMPILER=clang"

     

       34
       34
       +
           "-DCMAKE_CXX_COMPILER=clang++"

     

       35
       35
       +
         ];

     

       36
       36
       +
       

     

       37
       37
       +
         meta = with lib; {

     

       38
       38
       +
           description = "OpenCL compilation with clang compiler";

     

       39
       39
       +
           homepage = "https://github.com/RadeonOpenCompute/clang-ocl";

     

       40
       40
       +
           license = with licenses; [ mit ];

     

       41
       41
       +
           maintainers = with maintainers; [ Madouura ];

     

       42
       42
       +
           broken = rocmVersion != clang.version;

     

       43
       43
       +
         };

     

       44
       44
       +
       }

+91

pkgs/development/libraries/composable_kernel/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , hip

     

       7
       7
       +
       , openmp

     

       8
       8
       +
       , gtest ? null

     

       9
       9
       +
       , buildTests ? false

     

       10
       10
       +
       , buildExamples ? false

     

       11
       11
       +
       , gpuTargets ? null # gpuTargets = [ "gfx803" "gfx900" "gfx1030" ... ]

     

       12
       12
       +
       }:

     

       13
       13
       +
       

     

       14
       14
       +
       assert buildTests -> gtest != null;

     

       15
       15
       +
       

     

       16
       16
       +
       # Several tests seem to either not compile or have a race condition

     

       17
       17
       +
       # Undefined reference to symbol '_ZTIN7testing4TestE'

     

       18
       18
       +
       # Try removing this next update

     

       19
       19
       +
       assert buildTests == false;

     

       20
       20
       +
       

     

       21
       21
       +
       stdenv.mkDerivation rec {

     

       22
       22
       +
         pname = "composable_kernel";

     

       23
       23
       +
         version = "unstable-2022-11-02";

     

       24
       24
       +
       

     

       25
       25
       +
         outputs = [

     

       26
       26
       +
           "out"

     

       27
       27
       +
         ] ++ lib.optionals buildTests [

     

       28
       28
       +
           "test"

     

       29
       29
       +
         ] ++ lib.optionals buildExamples [

     

       30
       30
       +
           "example"

     

       31
       31
       +
         ];

     

       32
       32
       +
       

     

       33
       33
       +
         src = fetchFromGitHub {

     

       34
       34
       +
           owner = "ROCmSoftwarePlatform";

     

       35
       35
       +
           repo = "composable_kernel";

     

       36
       36
       +
           rev = "79aa3fb1793c265c59d392e916baa851a55521c8";

     

       37
       37
       +
           hash = "sha256-vIfMdvRYCTqrjMGSb7gQfodzLw2wf3tGoCAa5jtfbvw=";

     

       38
       38
       +
         };

     

       39
       39
       +
       

     

       40
       40
       +
         nativeBuildInputs = [

     

       41
       41
       +
           cmake

     

       42
       42
       +
           rocm-cmake

     

       43
       43
       +
           hip

     

       44
       44
       +
         ];

     

       45
       45
       +
       

     

       46
       46
       +
         buildInputs = [

     

       47
       47
       +
           openmp

     

       48
       48
       +
         ] ++ lib.optionals buildTests [

     

       49
       49
       +
           gtest

     

       50
       50
       +
         ];

     

       51
       51
       +
       

     

       52
       52
       +
         cmakeFlags = [

     

       53
       53
       +
           "-DCMAKE_C_COMPILER=hipcc"

     

       54
       54
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       55
       55
       +
         ] ++ lib.optionals (gpuTargets != null) [

     

       56
       56
       +
           "-DGPU_TARGETS=${lib.strings.concatStringsSep ";" gpuTargets}"

     

       57
       57
       +
         ];

     

       58
       58
       +
       

     

       59
       59
       +
         # No flags to build selectively it seems...

     

       60
       60
       +
         postPatch = ''

     

       61
       61
       +
           substituteInPlace test/CMakeLists.txt \

     

       62
       62
       +
             --replace "include(googletest)" ""

     

       63
       63
       +
       

     

       64
       64
       +
           substituteInPlace CMakeLists.txt \

     

       65
       65
       +
             --replace "enable_testing()" ""

     

       66
       66
       +
         '' + lib.optionalString (!buildTests) ''

     

       67
       67
       +
           substituteInPlace CMakeLists.txt \

     

       68
       68
       +
             --replace "add_subdirectory(test)" ""

     

       69
       69
       +
         '' + lib.optionalString (!buildExamples) ''

     

       70
       70
       +
           substituteInPlace CMakeLists.txt \

     

       71
       71
       +
             --replace "add_subdirectory(example)" ""

     

       72
       72
       +
         '';

     

       73
       73
       +
       

     

       74
       74
       +
         postInstall = ''

     

       75
       75
       +
           mkdir -p $out/bin

     

       76
       76
       +
           mv bin/ckProfiler $out/bin

     

       77
       77
       +
         '' + lib.optionalString buildTests ''

     

       78
       78
       +
           mkdir -p $test/bin

     

       79
       79
       +
           mv bin/test_* $test/bin

     

       80
       80
       +
         '' + lib.optionalString buildExamples ''

     

       81
       81
       +
           mkdir -p $example/bin

     

       82
       82
       +
           mv bin/example_* $example/bin

     

       83
       83
       +
         '';

     

       84
       84
       +
       

     

       85
       85
       +
         meta = with lib; {

     

       86
       86
       +
           description = "Performance portable programming model for machine learning tensor operators";

     

       87
       87
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/composable_kernel";

     

       88
       88
       +
           license = with licenses; [ mit ];

     

       89
       89
       +
           maintainers = with maintainers; [ Madouura ];

     

       90
       90
       +
         };

     

       91
       91
       +
       }

+89

pkgs/development/libraries/hipcub/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , rocprim

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , gtest ? null

     

       12
       12
       +
       , gbenchmark ? null

     

       13
       13
       +
       , buildTests ? false

     

       14
       14
       +
       , buildBenchmarks ? false

     

       15
       15
       +
       }:

     

       16
       16
       +
       

     

       17
       17
       +
       assert buildTests -> gtest != null;

     

       18
       18
       +
       assert buildBenchmarks -> gbenchmark != null;

     

       19
       19
       +
       

     

       20
       20
       +
       # CUB can also be used as a backend instead of rocPRIM.

     

       21
       21
       +
       stdenv.mkDerivation rec {

     

       22
       22
       +
         pname = "hipcub";

     

       23
       23
       +
         rocmVersion = "5.3.1";

     

       24
       24
       +
         version = "2.12.0-${rocmVersion}";

     

       25
       25
       +
       

     

       26
       26
       +
         outputs = [

     

       27
       27
       +
           "out"

     

       28
       28
       +
         ] ++ lib.optionals buildTests [

     

       29
       29
       +
           "test"

     

       30
       30
       +
         ] ++ lib.optionals buildBenchmarks [

     

       31
       31
       +
           "benchmark"

     

       32
       32
       +
         ];

     

       33
       33
       +
       

     

       34
       34
       +
         src = fetchFromGitHub {

     

       35
       35
       +
           owner = "ROCmSoftwarePlatform";

     

       36
       36
       +
           repo = "hipCUB";

     

       37
       37
       +
           rev = "rocm-${rocmVersion}";

     

       38
       38
       +
           hash = "sha256-/GMZKbMD1sZQCM2FulM9jiJQ8ByYZinn0C8d/deFh0g=";

     

       39
       39
       +
         };

     

       40
       40
       +
       

     

       41
       41
       +
         nativeBuildInputs = [

     

       42
       42
       +
           cmake

     

       43
       43
       +
           rocm-cmake

     

       44
       44
       +
           hip

     

       45
       45
       +
         ];

     

       46
       46
       +
       

     

       47
       47
       +
         buildInputs = [

     

       48
       48
       +
           rocm-runtime

     

       49
       49
       +
           rocm-device-libs

     

       50
       50
       +
           rocm-comgr

     

       51
       51
       +
           rocprim

     

       52
       52
       +
         ] ++ lib.optionals buildTests [

     

       53
       53
       +
           gtest

     

       54
       54
       +
         ] ++ lib.optionals buildBenchmarks [

     

       55
       55
       +
           gbenchmark

     

       56
       56
       +
         ];

     

       57
       57
       +
       

     

       58
       58
       +
         cmakeFlags = [

     

       59
       59
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       60
       60
       +
           "-DHIP_ROOT_DIR=${hip}"

     

       61
       61
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       62
       62
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       63
       63
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       64
       64
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       65
       65
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       66
       66
       +
         ] ++ lib.optionals buildTests [

     

       67
       67
       +
           "-DBUILD_TEST=ON"

     

       68
       68
       +
         ] ++ lib.optionals buildBenchmarks [

     

       69
       69
       +
           "-DBUILD_BENCHMARK=ON"

     

       70
       70
       +
         ];

     

       71
       71
       +
       

     

       72
       72
       +
         postInstall = lib.optionalString buildTests ''

     

       73
       73
       +
           mkdir -p $test/bin

     

       74
       74
       +
           mv $out/bin/test_* $test/bin

     

       75
       75
       +
         '' + lib.optionalString buildBenchmarks ''

     

       76
       76
       +
           mkdir -p $benchmark/bin

     

       77
       77
       +
           mv $out/bin/benchmark_* $benchmark/bin

     

       78
       78
       +
         '' + lib.optionalString (buildTests || buildBenchmarks) ''

     

       79
       79
       +
           rmdir $out/bin

     

       80
       80
       +
         '';

     

       81
       81
       +
       

     

       82
       82
       +
         meta = with lib; {

     

       83
       83
       +
           description = "Thin wrapper library on top of rocPRIM or CUB";

     

       84
       84
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/hipCUB";

     

       85
       85
       +
           license = with licenses; [ bsd3 ];

     

       86
       86
       +
           maintainers = with maintainers; [ Madouura ];

     

       87
       87
       +
           broken = rocmVersion != hip.version;

     

       88
       88
       +
         };

     

       89
       89
       +
       }

+132

pkgs/development/libraries/hipsparse/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , rocsparse

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , gfortran

     

       12
       12
       +
       , git

     

       13
       13
       +
       , fetchzip ? null

     

       14
       14
       +
       , gtest ? null

     

       15
       15
       +
       , buildTests ? false

     

       16
       16
       +
       }:

     

       17
       17
       +
       

     

       18
       18
       +
       assert buildTests -> fetchzip != null;

     

       19
       19
       +
       assert buildTests -> gtest != null;

     

       20
       20
       +
       

     

       21
       21
       +
       # This can also use cuSPARSE as a backend instead of rocSPARSE

     

       22
       22
       +
       let

     

       23
       23
       +
         matrices = lib.optionalAttrs buildTests import ./deps.nix {

     

       24
       24
       +
           inherit fetchzip;

     

       25
       25
       +
           mirror1 = "https://sparse.tamu.edu/MM";

     

       26
       26
       +
           mirror2 = "https://www.cise.ufl.edu/research/sparse/MM";

     

       27
       27
       +
         };

     

       28
       28
       +
       in stdenv.mkDerivation rec {

     

       29
       29
       +
         pname = "hipsparse";

     

       30
       30
       +
         rocmVersion = "5.3.1";

     

       31
       31
       +
         version = "2.3.1-${rocmVersion}";

     

       32
       32
       +
       

     

       33
       33
       +
         outputs = [

     

       34
       34
       +
           "out"

     

       35
       35
       +
         ] ++ lib.optionals buildTests [

     

       36
       36
       +
           "test"

     

       37
       37
       +
         ];

     

       38
       38
       +
       

     

       39
       39
       +
         src = fetchFromGitHub {

     

       40
       40
       +
           owner = "ROCmSoftwarePlatform";

     

       41
       41
       +
           repo = "hipSPARSE";

     

       42
       42
       +
           rev = "rocm-${rocmVersion}";

     

       43
       43
       +
           hash = "sha256-Phcihat774ZSAe1QetE/GSZzGlnCnvS9GwsHBHCaD4c=";

     

       44
       44
       +
         };

     

       45
       45
       +
       

     

       46
       46
       +
         nativeBuildInputs = [

     

       47
       47
       +
           cmake

     

       48
       48
       +
           rocm-cmake

     

       49
       49
       +
           hip

     

       50
       50
       +
           gfortran

     

       51
       51
       +
         ];

     

       52
       52
       +
       

     

       53
       53
       +
         buildInputs = [

     

       54
       54
       +
           rocm-runtime

     

       55
       55
       +
           rocm-device-libs

     

       56
       56
       +
           rocm-comgr

     

       57
       57
       +
           rocsparse

     

       58
       58
       +
           git

     

       59
       59
       +
         ] ++ lib.optionals buildTests [

     

       60
       60
       +
           gtest

     

       61
       61
       +
         ];

     

       62
       62
       +
       

     

       63
       63
       +
         cmakeFlags = [

     

       64
       64
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       65
       65
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       66
       66
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       67
       67
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       68
       68
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       69
       69
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       70
       70
       +
         ] ++ lib.optionals buildTests [

     

       71
       71
       +
           "-DBUILD_CLIENTS_TESTS=ON"

     

       72
       72
       +
         ];

     

       73
       73
       +
       

     

       74
       74
       +
         # We have to manually generate the matrices

     

       75
       75
       +
         # CMAKE_MATRICES_DIR seems to be reset in clients/tests/CMakeLists.txt

     

       76
       76
       +
         postPatch = ''

     

       77
       77
       +
           substituteInPlace clients/common/utility.cpp \

     

       78
       78
       +
             --replace "#ifdef __cpp_lib_filesystem" " #if true"

     

       79
       79
       +
         '' + lib.optionalString buildTests ''

     

       80
       80
       +
           mkdir -p matrices

     

       81
       81
       +
       

     

       82
       82
       +
           ln -s ${matrices.matrix-01}/*.mtx matrices

     

       83
       83
       +
           ln -s ${matrices.matrix-02}/*.mtx matrices

     

       84
       84
       +
           ln -s ${matrices.matrix-03}/*.mtx matrices

     

       85
       85
       +
           ln -s ${matrices.matrix-04}/*.mtx matrices

     

       86
       86
       +
           ln -s ${matrices.matrix-05}/*.mtx matrices

     

       87
       87
       +
           ln -s ${matrices.matrix-06}/*.mtx matrices

     

       88
       88
       +
           ln -s ${matrices.matrix-07}/*.mtx matrices

     

       89
       89
       +
           ln -s ${matrices.matrix-08}/*.mtx matrices

     

       90
       90
       +
           ln -s ${matrices.matrix-09}/*.mtx matrices

     

       91
       91
       +
           ln -s ${matrices.matrix-10}/*.mtx matrices

     

       92
       92
       +
           ln -s ${matrices.matrix-11}/*.mtx matrices

     

       93
       93
       +
           ln -s ${matrices.matrix-12}/*.mtx matrices

     

       94
       94
       +
           ln -s ${matrices.matrix-13}/*.mtx matrices

     

       95
       95
       +
           ln -s ${matrices.matrix-14}/*.mtx matrices

     

       96
       96
       +
           ln -s ${matrices.matrix-15}/*.mtx matrices

     

       97
       97
       +
           ln -s ${matrices.matrix-16}/*.mtx matrices

     

       98
       98
       +
           ln -s ${matrices.matrix-17}/*.mtx matrices

     

       99
       99
       +
           ln -s ${matrices.matrix-18}/*.mtx matrices

     

       100
       100
       +
           ln -s ${matrices.matrix-19}/*.mtx matrices

     

       101
       101
       +
       

     

       102
       102
       +
           # Not used by the original cmake, causes an error

     

       103
       103
       +
           rm matrices/*_b.mtx

     

       104
       104
       +
       

     

       105
       105
       +
           echo "deps/convert.cpp -> deps/mtx2csr"

     

       106
       106
       +
           hipcc deps/convert.cpp -O3 -o deps/mtx2csr

     

       107
       107
       +
       

     

       108
       108
       +
           for mat in $(ls -1 matrices | cut -d "." -f 1); do

     

       109
       109
       +
             echo "mtx2csr: $mat.mtx -> $mat.bin"

     

       110
       110
       +
             deps/mtx2csr matrices/$mat.mtx matrices/$mat.bin

     

       111
       111
       +
             unlink matrices/$mat.mtx

     

       112
       112
       +
           done

     

       113
       113
       +
       

     

       114
       114
       +
           substituteInPlace clients/tests/CMakeLists.txt \

     

       115
       115
       +
             --replace "\''${PROJECT_BINARY_DIR}/matrices" "/build/source/matrices"

     

       116
       116
       +
         '';

     

       117
       117
       +
       

     

       118
       118
       +
         postInstall = lib.optionalString buildTests ''

     

       119
       119
       +
           mkdir -p $test/bin

     

       120
       120
       +
           mv $out/bin/hipsparse-test $test/bin

     

       121
       121
       +
           mv /build/source/matrices $test

     

       122
       122
       +
           rmdir $out/bin

     

       123
       123
       +
         '';

     

       124
       124
       +
       

     

       125
       125
       +
         meta = with lib; {

     

       126
       126
       +
           description = "ROCm SPARSE marshalling library";

     

       127
       127
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/hipSPARSE";

     

       128
       128
       +
           license = with licenses; [ mit ];

     

       129
       129
       +
           maintainers = with maintainers; [ Madouura ];

     

       130
       130
       +
           broken = rocmVersion != hip.version;

     

       131
       131
       +
         };

     

       132
       132
       +
       }

+177

pkgs/development/libraries/hipsparse/deps.nix

···

       1
       1
       +
       { fetchzip

     

       2
       2
       +
       , mirror1

     

       3
       3
       +
       , mirror2

     

       4
       4
       +
       }:

     

       5
       5
       +
       

     

       6
       6
       +
       {

     

       7
       7
       +
         matrix-01 = fetchzip {

     

       8
       8
       +
           sha256 = "sha256-AHur5ZIDZTFRrO2GV0ieXrffq4KUiGWiZ59pv0fUtEQ=";

     

       9
       9
       +
       

     

       10
       10
       +
           urls = [

     

       11
       11
       +
             "${mirror1}/SNAP/amazon0312.tar.gz"

     

       12
       12
       +
             "${mirror2}/SNAP/amazon0312.tar.gz"

     

       13
       13
       +
           ];

     

       14
       14
       +
         };

     

       15
       15
       +
       

     

       16
       16
       +
         matrix-02 = fetchzip {

     

       17
       17
       +
           sha256 = "sha256-0rSxaN4lQcdaCLsvlgicG70FXUxXeERPiEmQ4MzbRdE=";

     

       18
       18
       +
       

     

       19
       19
       +
           urls = [

     

       20
       20
       +
             "${mirror1}/Muite/Chebyshev4.tar.gz"

     

       21
       21
       +
             "${mirror2}/Muite/Chebyshev4.tar.gz"

     

       22
       22
       +
           ];

     

       23
       23
       +
         };

     

       24
       24
       +
       

     

       25
       25
       +
         matrix-03 = fetchzip {

     

       26
       26
       +
           sha256 = "sha256-hDzDWDUnHEyFedX/tMNq83ZH8uWyM4xtZYUUAD3rizo=";

     

       27
       27
       +
       

     

       28
       28
       +
           urls = [

     

       29
       29
       +
             "${mirror1}/FEMLAB/sme3Dc.tar.gz"

     

       30
       30
       +
             "${mirror2}/FEMLAB/sme3Dc.tar.gz"

     

       31
       31
       +
           ];

     

       32
       32
       +
         };

     

       33
       33
       +
       

     

       34
       34
       +
         matrix-04 = fetchzip {

     

       35
       35
       +
           sha256 = "sha256-GmN2yOt/MoX01rKe05aTyB3ypUP4YbQGOITZ0BqPmC0=";

     

       36
       36
       +
       

     

       37
       37
       +
           urls = [

     

       38
       38
       +
             "${mirror1}/Williams/webbase-1M.tar.gz"

     

       39
       39
       +
             "${mirror2}/Williams/webbase-1M.tar.gz"

     

       40
       40
       +
           ];

     

       41
       41
       +
         };

     

       42
       42
       +
       

     

       43
       43
       +
         matrix-05 = fetchzip {

     

       44
       44
       +
           sha256 = "sha256-gQNjfVyWzNM9RwImJGhkhahRmZz74LzDs1oijL7mI7k=";

     

       45
       45
       +
       

     

       46
       46
       +
           urls = [

     

       47
       47
       +
             "${mirror1}/Williams/mac_econ_fwd500.tar.gz"

     

       48
       48
       +
             "${mirror2}/Williams/mac_econ_fwd500.tar.gz"

     

       49
       49
       +
           ];

     

       50
       50
       +
         };

     

       51
       51
       +
       

     

       52
       52
       +
         matrix-06 = fetchzip {

     

       53
       53
       +
           sha256 = "sha256-87cdZjntNcTuz5BtO59irhcuRbPllWSbhCEX3Td02qc=";

     

       54
       54
       +
       

     

       55
       55
       +
           urls = [

     

       56
       56
       +
             "${mirror1}/Williams/mc2depi.tar.gz"

     

       57
       57
       +
             "${mirror2}/Williams/mc2depi.tar.gz"

     

       58
       58
       +
           ];

     

       59
       59
       +
         };

     

       60
       60
       +
       

     

       61
       61
       +
         matrix-07 = fetchzip {

     

       62
       62
       +
           sha256 = "sha256-WRamuJX3D8Tm+k0q67RjUDG3DeNAxhKiaPkk5afY5eU=";

     

       63
       63
       +
       

     

       64
       64
       +
           urls = [

     

       65
       65
       +
             "${mirror1}/Bova/rma10.tar.gz"

     

       66
       66
       +
             "${mirror2}/Bova/rma10.tar.gz"

     

       67
       67
       +
           ];

     

       68
       68
       +
         };

     

       69
       69
       +
       

     

       70
       70
       +
         matrix-08 = fetchzip {

     

       71
       71
       +
           sha256 = "sha256-5dhkm293Mc3lzakKxHy5W5XIn4Rw+gihVh7gyrjEHXo=";

     

       72
       72
       +
       

     

       73
       73
       +
           urls = [

     

       74
       74
       +
             "${mirror1}/JGD_BIBD/bibd_22_8.tar.gz"

     

       75
       75
       +
             "${mirror2}/JGD_BIBD/bibd_22_8.tar.gz"

     

       76
       76
       +
           ];

     

       77
       77
       +
         };

     

       78
       78
       +
       

     

       79
       79
       +
         matrix-09 = fetchzip {

     

       80
       80
       +
           sha256 = "sha256-czjLWCjXAjZCk5TGYHaEkwSAzQu3TQ3QyB6eNKR4G88=";

     

       81
       81
       +
       

     

       82
       82
       +
           urls = [

     

       83
       83
       +
             "${mirror1}/Hamm/scircuit.tar.gz"

     

       84
       84
       +
             "${mirror2}/Hamm/scircuit.tar.gz"

     

       85
       85
       +
           ];

     

       86
       86
       +
         };

     

       87
       87
       +
       

     

       88
       88
       +
         matrix-10 = fetchzip {

     

       89
       89
       +
           sha256 = "sha256-bYuLnJViAIcIejAkh69/bsNAVIDU4wfTLtD+nmHd6FM=";

     

       90
       90
       +
       

     

       91
       91
       +
           urls = [

     

       92
       92
       +
             "${mirror1}/Sandia/ASIC_320k.tar.gz"

     

       93
       93
       +
             "${mirror2}/Sandia/ASIC_320k.tar.gz"

     

       94
       94
       +
           ];

     

       95
       95
       +
         };

     

       96
       96
       +
       

     

       97
       97
       +
         matrix-11 = fetchzip {

     

       98
       98
       +
           sha256 = "sha256-aDwn8P1khYjo2Agbq5m9ZBInJUxf/knJNvyptt0fak0=";

     

       99
       99
       +
       

     

       100
       100
       +
           urls = [

     

       101
       101
       +
             "${mirror1}/GHS_psdef/bmwcra_1.tar.gz"

     

       102
       102
       +
             "${mirror2}/GHS_psdef/bmwcra_1.tar.gz"

     

       103
       103
       +
           ];

     

       104
       104
       +
         };

     

       105
       105
       +
       

     

       106
       106
       +
         matrix-12 = fetchzip {

     

       107
       107
       +
           sha256 = "sha256-8OJqA/byhlAZd869TPUzZFdsOiwOoRGfKyhM+RMjXoY=";

     

       108
       108
       +
       

     

       109
       109
       +
           urls = [

     

       110
       110
       +
             "${mirror1}/HB/nos1.tar.gz"

     

       111
       111
       +
             "${mirror2}/HB/nos1.tar.gz"

     

       112
       112
       +
           ];

     

       113
       113
       +
         };

     

       114
       114
       +
       

     

       115
       115
       +
         matrix-13 = fetchzip {

     

       116
       116
       +
           sha256 = "sha256-FS0rKqmg+uHwsM/yGfQLBdd7LH/rUrdutkNGBD/Mh1I=";

     

       117
       117
       +
       

     

       118
       118
       +
           urls = [

     

       119
       119
       +
             "${mirror1}/HB/nos2.tar.gz"

     

       120
       120
       +
             "${mirror2}/HB/nos2.tar.gz"

     

       121
       121
       +
           ];

     

       122
       122
       +
         };

     

       123
       123
       +
       

     

       124
       124
       +
         matrix-14 = fetchzip {

     

       125
       125
       +
           sha256 = "sha256-DANnlrNJikrI7Pst9vRedtbuxepyHmCIu2yhltc4Qcs=";

     

       126
       126
       +
       

     

       127
       127
       +
           urls = [

     

       128
       128
       +
             "${mirror1}/HB/nos3.tar.gz"

     

       129
       129
       +
             "${mirror2}/HB/nos3.tar.gz"

     

       130
       130
       +
           ];

     

       131
       131
       +
         };

     

       132
       132
       +
       

     

       133
       133
       +
         matrix-15 = fetchzip {

     

       134
       134
       +
           sha256 = "sha256-21mUgqjWGUfYgiWwSrKh9vH8Vdt3xzcefmqYNYRpxiY=";

     

       135
       135
       +
       

     

       136
       136
       +
           urls = [

     

       137
       137
       +
             "${mirror1}/HB/nos4.tar.gz"

     

       138
       138
       +
             "${mirror2}/HB/nos4.tar.gz"

     

       139
       139
       +
           ];

     

       140
       140
       +
         };

     

       141
       141
       +
       

     

       142
       142
       +
         matrix-16 = fetchzip {

     

       143
       143
       +
           sha256 = "sha256-FOuXvGqBBFNkVS6cexmkluret54hCfCOdK+DOZllE4c=";

     

       144
       144
       +
       

     

       145
       145
       +
           urls = [

     

       146
       146
       +
             "${mirror1}/HB/nos5.tar.gz"

     

       147
       147
       +
             "${mirror2}/HB/nos5.tar.gz"

     

       148
       148
       +
           ];

     

       149
       149
       +
         };

     

       150
       150
       +
       

     

       151
       151
       +
         matrix-17 = fetchzip {

     

       152
       152
       +
           sha256 = "sha256-+7NI1rA/qQxYPpjXKHvAaCZ+LSaAJ4xuJvMRMBEUYxg=";

     

       153
       153
       +
       

     

       154
       154
       +
           urls = [

     

       155
       155
       +
             "${mirror1}/HB/nos6.tar.gz"

     

       156
       156
       +
             "${mirror2}/HB/nos6.tar.gz"

     

       157
       157
       +
           ];

     

       158
       158
       +
         };

     

       159
       159
       +
       

     

       160
       160
       +
         matrix-18 = fetchzip {

     

       161
       161
       +
           sha256 = "sha256-q3NxJjbwGGcFiQ9nhWfUKgZmdVwCfPmgQoqy0AqOsNc=";

     

       162
       162
       +
       

     

       163
       163
       +
           urls = [

     

       164
       164
       +
             "${mirror1}/HB/nos7.tar.gz"

     

       165
       165
       +
             "${mirror2}/HB/nos7.tar.gz"

     

       166
       166
       +
           ];

     

       167
       167
       +
         };

     

       168
       168
       +
       

     

       169
       169
       +
         matrix-19 = fetchzip {

     

       170
       170
       +
           sha256 = "sha256-0GAN6qmVfD+tprIigzuUUUwm5KVhkN9X65wMEvFltDY=";

     

       171
       171
       +
       

     

       172
       172
       +
           urls = [

     

       173
       173
       +
             "${mirror1}/DNVS/shipsec1.tar.gz"

     

       174
       174
       +
             "${mirror2}/DNVS/shipsec1.tar.gz"

     

       175
       175
       +
           ];

     

       176
       176
       +
         };

     

       177
       177
       +
       }

+129

pkgs/development/libraries/miopengemm/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-opencl-runtime

     

       7
       7
       +
       , clang

     

       8
       8
       +
       , texlive ? null

     

       9
       9
       +
       , doxygen ? null

     

       10
       10
       +
       , sphinx ? null

     

       11
       11
       +
       , python3Packages ? null

     

       12
       12
       +
       , openblas ? null

     

       13
       13
       +
       , buildDocs ? false

     

       14
       14
       +
       , buildTests ? false

     

       15
       15
       +
       , buildBenchmarks ? false

     

       16
       16
       +
       }:

     

       17
       17
       +
       

     

       18
       18
       +
       assert buildDocs -> texlive != null;

     

       19
       19
       +
       assert buildDocs -> doxygen != null;

     

       20
       20
       +
       assert buildDocs -> sphinx != null;

     

       21
       21
       +
       assert buildDocs -> python3Packages != null;

     

       22
       22
       +
       assert buildTests -> openblas != null;

     

       23
       23
       +
       

     

       24
       24
       +
       let

     

       25
       25
       +
         latex = lib.optionalAttrs buildDocs (texlive.combine {

     

       26
       26
       +
           inherit (texlive) scheme-small

     

       27
       27
       +
           latexmk

     

       28
       28
       +
           tex-gyre

     

       29
       29
       +
           fncychap

     

       30
       30
       +
           wrapfig

     

       31
       31
       +
           capt-of

     

       32
       32
       +
           framed

     

       33
       33
       +
           needspace

     

       34
       34
       +
           tabulary

     

       35
       35
       +
           varwidth

     

       36
       36
       +
           titlesec;

     

       37
       37
       +
         });

     

       38
       38
       +
       in stdenv.mkDerivation rec {

     

       39
       39
       +
         pname = "miopengemm";

     

       40
       40
       +
         rocmVersion = "5.3.1";

     

       41
       41
       +
         version = rocmVersion;

     

       42
       42
       +
       

     

       43
       43
       +
         outputs = [

     

       44
       44
       +
           "out"

     

       45
       45
       +
         ] ++ lib.optionals buildDocs [

     

       46
       46
       +
           "docs"

     

       47
       47
       +
         ] ++ lib.optionals buildTests [

     

       48
       48
       +
           "test"

     

       49
       49
       +
         ] ++ lib.optionals buildBenchmarks [

     

       50
       50
       +
           "benchmark"

     

       51
       51
       +
         ];

     

       52
       52
       +
       

     

       53
       53
       +
         src = fetchFromGitHub {

     

       54
       54
       +
           owner = "ROCmSoftwarePlatform";

     

       55
       55
       +
           repo = "MIOpenGEMM";

     

       56
       56
       +
           rev = "rocm-${rocmVersion}";

     

       57
       57
       +
           hash = "sha256-AiRzOMYRA/0nbQomyq4oOEwNZdkPYWRA2W6QFlctvFc=";

     

       58
       58
       +
         };

     

       59
       59
       +
       

     

       60
       60
       +
         nativeBuildInputs = [

     

       61
       61
       +
           cmake

     

       62
       62
       +
           rocm-cmake

     

       63
       63
       +
           clang

     

       64
       64
       +
         ];

     

       65
       65
       +
       

     

       66
       66
       +
         buildInputs = [

     

       67
       67
       +
           rocm-opencl-runtime

     

       68
       68
       +
         ] ++ lib.optionals buildDocs [

     

       69
       69
       +
           latex

     

       70
       70
       +
           doxygen

     

       71
       71
       +
           sphinx

     

       72
       72
       +
           python3Packages.sphinx_rtd_theme

     

       73
       73
       +
           python3Packages.breathe

     

       74
       74
       +
         ] ++ lib.optionals buildTests [

     

       75
       75
       +
           openblas

     

       76
       76
       +
         ];

     

       77
       77
       +
       

     

       78
       78
       +
         cmakeFlags = [

     

       79
       79
       +
           "-DCMAKE_C_COMPILER=clang"

     

       80
       80
       +
           "-DCMAKE_CXX_COMPILER=clang++"

     

       81
       81
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       82
       82
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       83
       83
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       84
       84
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       85
       85
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       86
       86
       +
         ] ++ lib.optionals buildTests [

     

       87
       87
       +
           "-DOPENBLAS=ON"

     

       88
       88
       +
         ] ++ lib.optionals buildBenchmarks [

     

       89
       89
       +
           "-DAPI_BENCH_MIOGEMM=ON"

     

       90
       90
       +
           # Needs https://github.com/CNugteren/CLBlast

     

       91
       91
       +
           # "-DAPI_BENCH_CLBLAST=ON"

     

       92
       92
       +
           # Needs https://github.com/openai/triton

     

       93
       93
       +
           # "-DAPI_BENCH_ISAAC=ON"

     

       94
       94
       +
         ];

     

       95
       95
       +
       

     

       96
       96
       +
         # Unfortunately, it seems like we have to call make on these manually

     

       97
       97
       +
         postBuild = lib.optionalString buildDocs ''

     

       98
       98
       +
           export HOME=$(mktemp -d)

     

       99
       99
       +
           make doc

     

       100
       100
       +
         '' + lib.optionalString buildTests ''

     

       101
       101
       +
           make check

     

       102
       102
       +
         '' + lib.optionalString buildBenchmarks ''

     

       103
       103
       +
           make examples

     

       104
       104
       +
         '';

     

       105
       105
       +
       

     

       106
       106
       +
         postInstall = lib.optionalString buildTests ''

     

       107
       107
       +
           mkdir -p $test/bin

     

       108
       108
       +
           find tests -executable -type f -exec mv {} $test/bin \;

     

       109
       109
       +
           patchelf --set-rpath ${lib.makeLibraryPath buildInputs}:$out/lib $test/bin/*

     

       110
       110
       +
         '' + lib.optionalString buildBenchmarks ''

     

       111
       111
       +
           mkdir -p $benchmark/bin

     

       112
       112
       +
           find examples -executable -type f -exec mv {} $benchmark/bin \;

     

       113
       113
       +
           patchelf --set-rpath ${lib.makeLibraryPath buildInputs}:$out/lib $benchmark/bin/*

     

       114
       114
       +
         '';

     

       115
       115
       +
       

     

       116
       116
       +
         postFixup = lib.optionalString buildDocs ''

     

       117
       117
       +
           mkdir -p $docs/share/doc/miopengemm

     

       118
       118
       +
           mv ../doc/html $docs/share/doc/miopengemm

     

       119
       119
       +
           mv ../doc/pdf/miopengemm.pdf $docs/share/doc/miopengemm

     

       120
       120
       +
         '';

     

       121
       121
       +
       

     

       122
       122
       +
         meta = with lib; {

     

       123
       123
       +
           description = "OpenCL general matrix multiplication API for ROCm";

     

       124
       124
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/MIOpenGEMM";

     

       125
       125
       +
           license = with licenses; [ mit ];

     

       126
       126
       +
           maintainers = with maintainers; [ Madouura ];

     

       127
       127
       +
           broken = rocmVersion != clang.version;

     

       128
       128
       +
         };

     

       129
       129
       +
       }

+84

pkgs/development/libraries/rccl/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , rocm-smi

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , gtest

     

       12
       12
       +
       , chrpath ? null

     

       13
       13
       +
       , buildTests ? false

     

       14
       14
       +
       }:

     

       15
       15
       +
       

     

       16
       16
       +
       assert buildTests -> chrpath != null;

     

       17
       17
       +
       

     

       18
       18
       +
       stdenv.mkDerivation rec {

     

       19
       19
       +
         pname = "rccl";

     

       20
       20
       +
         rocmVersion = "5.3.1";

     

       21
       21
       +
         version = "2.12.10-${rocmVersion}";

     

       22
       22
       +
       

     

       23
       23
       +
         outputs = [

     

       24
       24
       +
           "out"

     

       25
       25
       +
         ] ++ lib.optionals buildTests [

     

       26
       26
       +
           "test"

     

       27
       27
       +
         ];

     

       28
       28
       +
       

     

       29
       29
       +
         src = fetchFromGitHub {

     

       30
       30
       +
           owner = "ROCmSoftwarePlatform";

     

       31
       31
       +
           repo = "rccl";

     

       32
       32
       +
           rev = "rocm-${rocmVersion}";

     

       33
       33
       +
           hash = "sha256-whRXGD8oINDYhFs8+hEWKWoGNqacGlyy7xi8peA8Qsk=";

     

       34
       34
       +
         };

     

       35
       35
       +
       

     

       36
       36
       +
         nativeBuildInputs = [

     

       37
       37
       +
           cmake

     

       38
       38
       +
           rocm-cmake

     

       39
       39
       +
           hip

     

       40
       40
       +
         ];

     

       41
       41
       +
       

     

       42
       42
       +
         buildInputs = [

     

       43
       43
       +
           rocm-runtime

     

       44
       44
       +
           rocm-device-libs

     

       45
       45
       +
           rocm-comgr

     

       46
       46
       +
           rocm-smi

     

       47
       47
       +
           gtest

     

       48
       48
       +
         ] ++ lib.optionals buildTests [

     

       49
       49
       +
           chrpath

     

       50
       50
       +
         ];

     

       51
       51
       +
       

     

       52
       52
       +
         cmakeFlags = [

     

       53
       53
       +
           "-DCMAKE_C_COMPILER=hipcc"

     

       54
       54
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       55
       55
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       56
       56
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       57
       57
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       58
       58
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       59
       59
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       60
       60
       +
         ] ++ lib.optionals buildTests [

     

       61
       61
       +
           "-DBUILD_TESTS=ON"

     

       62
       62
       +
         ];

     

       63
       63
       +
       

     

       64
       64
       +
         # Replace the manually set parallel jobs to NIX_BUILD_CORES

     

       65
       65
       +
         postPatch = ''

     

       66
       66
       +
           substituteInPlace CMakeLists.txt \

     

       67
       67
       +
             --replace "8 P" "$NIX_BUILD_CORES P" \

     

       68
       68
       +
             --replace "8)" "$NIX_BUILD_CORES)"

     

       69
       69
       +
         '';

     

       70
       70
       +
       

     

       71
       71
       +
         postInstall = lib.optionalString buildTests ''

     

       72
       72
       +
           mkdir -p $test/bin

     

       73
       73
       +
           mv $out/bin/* $test/bin

     

       74
       74
       +
           rmdir $out/bin

     

       75
       75
       +
         '';

     

       76
       76
       +
       

     

       77
       77
       +
         meta = with lib; {

     

       78
       78
       +
           description = "ROCm communication collectives library";

     

       79
       79
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rccl";

     

       80
       80
       +
           license = with licenses; [ bsd2 bsd3 ];

     

       81
       81
       +
           maintainers = with maintainers; [ Madouura ];

     

       82
       82
       +
           broken = rocmVersion != hip.version;

     

       83
       83
       +
         };

     

       84
       84
       +
       }

+136

pkgs/development/libraries/rocblas/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , fetchpatch

     

       5
       5
       +
       , cmake

     

       6
       6
       +
       , rocm-cmake

     

       7
       7
       +
       , rocm-runtime

     

       8
       8
       +
       , rocm-device-libs

     

       9
       9
       +
       , rocm-comgr

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , python3

     

       12
       12
       +
       , tensile ? null

     

       13
       13
       +
       , msgpack ? null

     

       14
       14
       +
       , libxml2 ? null

     

       15
       15
       +
       , llvm ? null

     

       16
       16
       +
       , python3Packages ? null

     

       17
       17
       +
       , gtest ? null

     

       18
       18
       +
       , gfortran ? null

     

       19
       19
       +
       , buildTensile ? true

     

       20
       20
       +
       , buildTests ? false

     

       21
       21
       +
       , buildBenchmarks ? false

     

       22
       22
       +
       , tensileLogic ? "asm_full"

     

       23
       23
       +
       , tensileCOVersion ? "V3"

     

       24
       24
       +
       , tensileSepArch ? true

     

       25
       25
       +
       , tensileLazyLib ? true

     

       26
       26
       +
       , tensileLibFormat ? "msgpack"

     

       27
       27
       +
       , gpuTargets ? [ "all" ]

     

       28
       28
       +
       }:

     

       29
       29
       +
       

     

       30
       30
       +
       assert buildTensile -> tensile != null;

     

       31
       31
       +
       assert buildTensile -> msgpack != null;

     

       32
       32
       +
       assert buildTensile -> libxml2 != null;

     

       33
       33
       +
       assert buildTensile -> llvm != null;

     

       34
       34
       +
       assert buildTensile -> python3Packages != null;

     

       35
       35
       +
       assert buildTests -> gtest != null;

     

       36
       36
       +
       assert buildTests -> gfortran != null;

     

       37
       37
       +
       

     

       38
       38
       +
       # Tests and benchmarks are a can of worms that I will tackle in a different PR

     

       39
       39
       +
       # It involves completely rewriting the amd-blis derivation

     

       40
       40
       +
       assert buildTests == false;

     

       41
       41
       +
       assert buildBenchmarks == false;

     

       42
       42
       +
       

     

       43
       43
       +
       stdenv.mkDerivation rec {

     

       44
       44
       +
         pname = "rocblas";

     

       45
       45
       +
         rocmVersion = "5.3.1";

     

       46
       46
       +
         version = "2.45.0-${rocmVersion}";

     

       47
       47
       +
       

     

       48
       48
       +
         src = fetchFromGitHub {

     

       49
       49
       +
           owner = "ROCmSoftwarePlatform";

     

       50
       50
       +
           repo = "rocBLAS";

     

       51
       51
       +
           rev = "rocm-${rocmVersion}";

     

       52
       52
       +
           hash = "sha256-GeeICEI1dNE6D+nUUlBtUncLkPowAa5n+bsy160EtaU=";

     

       53
       53
       +
         };

     

       54
       54
       +
       

     

       55
       55
       +
         # We currently need this patch due to faulty toolchain includes

     

       56
       56
       +
         # See: https://github.com/ROCmSoftwarePlatform/rocBLAS/issues/1277

     

       57
       57
       +
         patches = [

     

       58
       58
       +
           (fetchpatch {

     

       59
       59
       +
             name = "only-std_norm-from-rocblas_complex.patch";

     

       60
       60
       +
             url = "https://github.com/ROCmSoftwarePlatform/rocBLAS/commit/44b99c6df26002139ca9ec68ee1fc8899c7b001f.patch";

     

       61
       61
       +
             hash = "sha256-vSZkVYY951fqfOThKFqnYBasWMblS6peEJZ6sFMCk9k=";

     

       62
       62
       +
           })

     

       63
       63
       +
         ];

     

       64
       64
       +
       

     

       65
       65
       +
         nativeBuildInputs = [

     

       66
       66
       +
           cmake

     

       67
       67
       +
           rocm-cmake

     

       68
       68
       +
           hip

     

       69
       69
       +
         ];

     

       70
       70
       +
       

     

       71
       71
       +
         buildInputs = [

     

       72
       72
       +
           rocm-runtime

     

       73
       73
       +
           rocm-device-libs

     

       74
       74
       +
           rocm-comgr

     

       75
       75
       +
           python3

     

       76
       76
       +
         ] ++ lib.optionals buildTensile [

     

       77
       77
       +
           msgpack

     

       78
       78
       +
           libxml2

     

       79
       79
       +
           llvm

     

       80
       80
       +
           python3Packages.pyyaml

     

       81
       81
       +
           python3Packages.msgpack

     

       82
       82
       +
         ] ++ lib.optionals buildTests [

     

       83
       83
       +
           gtest

     

       84
       84
       +
           gfortran

     

       85
       85
       +
         ];

     

       86
       86
       +
       

     

       87
       87
       +
         cmakeFlags = [

     

       88
       88
       +
           "-DCMAKE_C_COMPILER=hipcc"

     

       89
       89
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       90
       90
       +
           "-Dpython=python3"

     

       91
       91
       +
           "-DAMDGPU_TARGETS=${lib.strings.concatStringsSep ";" gpuTargets}"

     

       92
       92
       +
           "-DBUILD_WITH_TENSILE=${if buildTensile then "ON" else "OFF"}"

     

       93
       93
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       94
       94
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       95
       95
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       96
       96
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       97
       97
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       98
       98
       +
         ] ++ lib.optionals buildTensile [

     

       99
       99
       +
           "-DVIRTUALENV_HOME_DIR=/build/source/tensile"

     

       100
       100
       +
           "-DTensile_TEST_LOCAL_PATH=/build/source/tensile"

     

       101
       101
       +
           "-DTensile_ROOT=/build/source/tensile/lib/python${python3.pythonVersion}/site-packages/Tensile"

     

       102
       102
       +
           "-DTensile_LOGIC=${tensileLogic}"

     

       103
       103
       +
           "-DTensile_CODE_OBJECT_VERSION=${tensileCOVersion}"

     

       104
       104
       +
           "-DTensile_SEPARATE_ARCHITECTURES=${if tensileSepArch then "ON" else "OFF"}"

     

       105
       105
       +
           "-DTensile_LAZY_LIBRARY_LOADING=${if tensileLazyLib then "ON" else "OFF"}"

     

       106
       106
       +
           "-DTensile_LIBRARY_FORMAT=${tensileLibFormat}"

     

       107
       107
       +
         ] ++ lib.optionals buildTests [

     

       108
       108
       +
           "-DBUILD_CLIENTS_TESTS=ON"

     

       109
       109
       +
         ] ++ lib.optionals buildBenchmarks [

     

       110
       110
       +
           "-DBUILD_CLIENTS_BENCHMARKS=ON"

     

       111
       111
       +
         ];

     

       112
       112
       +
       

     

       113
       113
       +
         # Tensile REALLY wants to write to the nix directory if we include it normally

     

       114
       114
       +
         # We need to manually fixup the path so tensile will generate .co and .dat files

     

       115
       115
       +
         postPatch = lib.optionalString buildTensile ''

     

       116
       116
       +
           export PATH=${llvm}/bin:$PATH

     

       117
       117
       +
           cp -a ${tensile} tensile

     

       118
       118
       +
           chmod +w -R tensile

     

       119
       119
       +
       

     

       120
       120
       +
           # Rewrap Tensile

     

       121
       121
       +
           substituteInPlace tensile/bin/{.t*,.T*,*} \

     

       122
       122
       +
             --replace "${tensile}" "/build/source/tensile"

     

       123
       123
       +
       

     

       124
       124
       +
           substituteInPlace CMakeLists.txt \

     

       125
       125
       +
             --replace "include(virtualenv)" "" \

     

       126
       126
       +
             --replace "virtualenv_install(\''${Tensile_TEST_LOCAL_PATH})" ""

     

       127
       127
       +
         '';

     

       128
       128
       +
       

     

       129
       129
       +
         meta = with lib; {

     

       130
       130
       +
           description = "BLAS implementation for ROCm platform";

     

       131
       131
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rocBLAS";

     

       132
       132
       +
           license = with licenses; [ mit ];

     

       133
       133
       +
           maintainers = with maintainers; [ Madouura ];

     

       134
       134
       +
           broken = rocmVersion != hip.version;

     

       135
       135
       +
         };

     

       136
       136
       +
       }

+2 -2

pkgs/development/libraries/rocclr/default.nix

···

       7
       7
        
       

     

       8
       8
        
       stdenv.mkDerivation rec {

     

       9
       9
        
         pname = "rocclr";

     

       10
       10
       -
         version = "5.3.0";

     

       10
       10
       +
         version = "5.3.1";

     

       11
       11
        
       

     

       12
       12
        
         src = fetchFromGitHub {

     

       13
       13
        
           owner = "ROCm-Developer-Tools";

     

       14
       14
        
           repo = "ROCclr";

     

       15
       15
        
           rev = "rocm-${version}";

     

       16
       16
       -
           hash = "sha256-l14+l8FkiFmGuRZ9dyD/PEYH9nHVRRg1vMXMnVhg3K4=";

     

       16
       16
       +
           hash = "sha256-dmL9krI/gHGQdOZ53+bQ7WjKcmJ+fZZP0lzF8ITLT4E=";

     

       17
       17
        
         };

     

       18
       18
        
       

     

       19
       19
        
         patches = [

+114

pkgs/development/libraries/rocfft/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , hip

     

       10
       10
       +
       , sqlite

     

       11
       11
       +
       , python3

     

       12
       12
       +
       , gtest ? null

     

       13
       13
       +
       , boost ? null

     

       14
       14
       +
       , fftw ? null

     

       15
       15
       +
       , fftwFloat ? null

     

       16
       16
       +
       , llvmPackages ? null

     

       17
       17
       +
       , buildTests ? false

     

       18
       18
       +
       , buildBenchmarks ? false

     

       19
       19
       +
       }:

     

       20
       20
       +
       

     

       21
       21
       +
       assert buildTests -> gtest != null;

     

       22
       22
       +
       assert buildBenchmarks -> fftw != null;

     

       23
       23
       +
       assert buildBenchmarks -> fftwFloat != null;

     

       24
       24
       +
       assert (buildTests || buildBenchmarks) -> boost != null;

     

       25
       25
       +
       assert (buildTests || buildBenchmarks) -> llvmPackages != null;

     

       26
       26
       +
       

     

       27
       27
       +
       stdenv.mkDerivation rec {

     

       28
       28
       +
         pname = "rocfft";

     

       29
       29
       +
         rocmVersion = "5.3.1";

     

       30
       30
       +
         version = "1.0.18-${rocmVersion}";

     

       31
       31
       +
       

     

       32
       32
       +
         outputs = [

     

       33
       33
       +
           "out"

     

       34
       34
       +
         ] ++ lib.optionals buildTests [

     

       35
       35
       +
           "test"

     

       36
       36
       +
         ] ++ lib.optionals buildBenchmarks [

     

       37
       37
       +
           "benchmark"

     

       38
       38
       +
         ];

     

       39
       39
       +
       

     

       40
       40
       +
         src = fetchFromGitHub {

     

       41
       41
       +
           owner = "ROCmSoftwarePlatform";

     

       42
       42
       +
           repo = "rocFFT";

     

       43
       43
       +
           rev = "rocm-${rocmVersion}";

     

       44
       44
       +
           hash = "sha256-jb2F1fRe+YLloYJ/KtzrptUDhmdBDBtddeW/g55owKM=";

     

       45
       45
       +
         };

     

       46
       46
       +
       

     

       47
       47
       +
         nativeBuildInputs = [

     

       48
       48
       +
           cmake

     

       49
       49
       +
           rocm-cmake

     

       50
       50
       +
           hip

     

       51
       51
       +
         ];

     

       52
       52
       +
       

     

       53
       53
       +
         buildInputs = [

     

       54
       54
       +
           rocm-runtime

     

       55
       55
       +
           rocm-device-libs

     

       56
       56
       +
           rocm-comgr

     

       57
       57
       +
           sqlite

     

       58
       58
       +
           python3

     

       59
       59
       +
         ] ++ lib.optionals buildTests [

     

       60
       60
       +
           gtest

     

       61
       61
       +
         ] ++ lib.optionals (buildTests || buildBenchmarks) [

     

       62
       62
       +
           boost

     

       63
       63
       +
           fftw

     

       64
       64
       +
           fftwFloat

     

       65
       65
       +
           llvmPackages.openmp

     

       66
       66
       +
         ];

     

       67
       67
       +
       

     

       68
       68
       +
         propogatedBuildInputs = lib.optionals buildTests [

     

       69
       69
       +
           fftw

     

       70
       70
       +
           fftwFloat

     

       71
       71
       +
         ];

     

       72
       72
       +
       

     

       73
       73
       +
         cmakeFlags = [

     

       74
       74
       +
           "-DCMAKE_C_COMPILER=hipcc"

     

       75
       75
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       76
       76
       +
           "-DUSE_HIP_CLANG=ON"

     

       77
       77
       +
           "-DSQLITE_USE_SYSTEM_PACKAGE=ON"

     

       78
       78
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       79
       79
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       80
       80
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       81
       81
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       82
       82
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       83
       83
       +
         ] ++ lib.optionals buildTests [

     

       84
       84
       +
           "-DBUILD_CLIENTS_TESTS=ON"

     

       85
       85
       +
         ] ++ lib.optionals buildBenchmarks [

     

       86
       86
       +
           "-DBUILD_CLIENTS_RIDER=ON"

     

       87
       87
       +
           "-DBUILD_CLIENTS_SAMPLES=ON"

     

       88
       88
       +
         ];

     

       89
       89
       +
       

     

       90
       90
       +
         postInstall = lib.optionalString buildTests ''

     

       91
       91
       +
           mkdir -p $test/{bin,lib/fftw}

     

       92
       92
       +
           cp -a $out/bin/* $test/bin

     

       93
       93
       +
           ln -s ${fftw}/lib/libfftw*.so $test/lib/fftw

     

       94
       94
       +
           ln -s ${fftwFloat}/lib/libfftw*.so $test/lib/fftw

     

       95
       95
       +
           rm -r $out/lib/fftw

     

       96
       96
       +
           rm $test/bin/{rocfft_rtc_helper,*-rider} || true

     

       97
       97
       +
         '' + lib.optionalString buildBenchmarks ''

     

       98
       98
       +
           mkdir -p $benchmark/bin

     

       99
       99
       +
           cp -a $out/bin/* $benchmark/bin

     

       100
       100
       +
           rm $benchmark/bin/{rocfft_rtc_helper,*-test} || true

     

       101
       101
       +
         '' + lib.optionalString (buildTests || buildBenchmarks ) ''

     

       102
       102
       +
           mv $out/bin/rocfft_rtc_helper $out

     

       103
       103
       +
           rm -r $out/bin/*

     

       104
       104
       +
           mv $out/rocfft_rtc_helper $out/bin

     

       105
       105
       +
         '';

     

       106
       106
       +
       

     

       107
       107
       +
         meta = with lib; {

     

       108
       108
       +
           description = "FFT implementation for ROCm ";

     

       109
       109
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rocFFT";

     

       110
       110
       +
           license = with licenses; [ mit ];

     

       111
       111
       +
           maintainers = with maintainers; [ Madouura ];

     

       112
       112
       +
           broken = rocmVersion != hip.version;

     

       113
       113
       +
         };

     

       114
       114
       +
       }

+1 -1

pkgs/development/libraries/rocm-comgr/default.nix

···

       2
       2
        
       

     

       3
       3
        
       stdenv.mkDerivation rec {

     

       4
       4
        
         pname = "rocm-comgr";

     

       5
       5
       -
         version = "5.3.0";

     

       5
       5
       +
         version = "5.3.1";

     

       6
       6
        
       

     

       7
       7
        
         src = fetchFromGitHub {

     

       8
       8
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/development/libraries/rocm-device-libs/default.nix

···

       8
       8
        
       

     

       9
       9
        
       stdenv.mkDerivation rec {

     

       10
       10
        
         pname = "rocm-device-libs";

     

       11
       11
       -
         version = "5.3.0";

     

       11
       11
       +
         version = "5.3.1";

     

       12
       12
        
       

     

       13
       13
        
         src = fetchFromGitHub {

     

       14
       14
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/development/libraries/rocm-opencl-runtime/default.nix

···

       22
       22
        
       

     

       23
       23
        
       stdenv.mkDerivation rec {

     

       24
       24
        
         pname = "rocm-opencl-runtime";

     

       25
       25
       -
         version = "5.3.0";

     

       25
       25
       +
         version = "5.3.1";

     

       26
       26
        
       

     

       27
       27
        
         src = fetchFromGitHub {

     

       28
       28
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/development/libraries/rocm-runtime/default.nix

···

       15
       15
        
       

     

       16
       16
        
       stdenv.mkDerivation rec {

     

       17
       17
        
         pname = "rocm-runtime";

     

       18
       18
       -
         version = "5.3.0";

     

       18
       18
       +
         version = "5.3.1";

     

       19
       19
        
       

     

       20
       20
        
         src = fetchFromGitHub {

     

       21
       21
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/development/libraries/rocm-thunk/default.nix

···

       9
       9
        
       

     

       10
       10
        
       stdenv.mkDerivation rec {

     

       11
       11
        
         pname = "rocm-thunk";

     

       12
       12
       -
         version = "5.3.0";

     

       12
       12
       +
         version = "5.3.1";

     

       13
       13
        
       

     

       14
       14
        
         src = fetchFromGitHub {

     

       15
       15
        
           owner = "RadeonOpenCompute";

+85

pkgs/development/libraries/rocprim/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , hip

     

       10
       10
       +
       , gtest ? null

     

       11
       11
       +
       , gbenchmark ? null

     

       12
       12
       +
       , buildTests ? false

     

       13
       13
       +
       , buildBenchmarks ? false

     

       14
       14
       +
       }:

     

       15
       15
       +
       

     

       16
       16
       +
       assert buildTests -> gtest != null;

     

       17
       17
       +
       assert buildBenchmarks -> gbenchmark != null;

     

       18
       18
       +
       

     

       19
       19
       +
       stdenv.mkDerivation rec {

     

       20
       20
       +
         pname = "rocprim";

     

       21
       21
       +
         rocmVersion = "5.3.1";

     

       22
       22
       +
         version = "2.11.0-${rocmVersion}";

     

       23
       23
       +
       

     

       24
       24
       +
         outputs = [

     

       25
       25
       +
           "out"

     

       26
       26
       +
         ] ++ lib.optionals buildTests [

     

       27
       27
       +
           "test"

     

       28
       28
       +
         ] ++ lib.optionals buildBenchmarks [

     

       29
       29
       +
           "benchmark"

     

       30
       30
       +
         ];

     

       31
       31
       +
       

     

       32
       32
       +
         src = fetchFromGitHub {

     

       33
       33
       +
           owner = "ROCmSoftwarePlatform";

     

       34
       34
       +
           repo = "rocPRIM";

     

       35
       35
       +
           rev = "rocm-${rocmVersion}";

     

       36
       36
       +
           hash = "sha256-aapvj9bwwlg7VJfnH1PVR8DulMcJh1xR6B4rPPGU6Q4=";

     

       37
       37
       +
         };

     

       38
       38
       +
       

     

       39
       39
       +
         nativeBuildInputs = [

     

       40
       40
       +
           cmake

     

       41
       41
       +
           rocm-cmake

     

       42
       42
       +
           hip

     

       43
       43
       +
         ];

     

       44
       44
       +
       

     

       45
       45
       +
         buildInputs = [

     

       46
       46
       +
           rocm-runtime

     

       47
       47
       +
           rocm-device-libs

     

       48
       48
       +
           rocm-comgr

     

       49
       49
       +
         ] ++ lib.optionals buildTests [

     

       50
       50
       +
           gtest

     

       51
       51
       +
         ] ++ lib.optionals buildBenchmarks [

     

       52
       52
       +
           gbenchmark

     

       53
       53
       +
         ];

     

       54
       54
       +
       

     

       55
       55
       +
         cmakeFlags = [

     

       56
       56
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       57
       57
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       58
       58
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       59
       59
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       60
       60
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       61
       61
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       62
       62
       +
         ] ++ lib.optionals buildTests [

     

       63
       63
       +
           "-DBUILD_TEST=ON"

     

       64
       64
       +
         ] ++ lib.optionals buildBenchmarks [

     

       65
       65
       +
           "-DBUILD_BENCHMARK=ON"

     

       66
       66
       +
         ];

     

       67
       67
       +
       

     

       68
       68
       +
         postInstall = lib.optionalString buildTests ''

     

       69
       69
       +
           mkdir -p $test/bin

     

       70
       70
       +
           mv $out/bin/test_* $test/bin

     

       71
       71
       +
         '' + lib.optionalString buildBenchmarks ''

     

       72
       72
       +
           mkdir -p $benchmark/bin

     

       73
       73
       +
           mv $out/bin/benchmark_* $benchmark/bin

     

       74
       74
       +
         '' + lib.optionalString (buildTests || buildBenchmarks) ''

     

       75
       75
       +
           rmdir $out/bin

     

       76
       76
       +
         '';

     

       77
       77
       +
       

     

       78
       78
       +
         meta = with lib; {

     

       79
       79
       +
           description = "ROCm parallel primitives";

     

       80
       80
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rocPRIM";

     

       81
       81
       +
           license = with licenses; [ mit ];

     

       82
       82
       +
           maintainers = with maintainers; [ Madouura ];

     

       83
       83
       +
           broken = rocmVersion != hip.version;

     

       84
       84
       +
         };

     

       85
       85
       +
       }

+148

pkgs/development/libraries/rocsparse/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , rocprim

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , gfortran

     

       12
       12
       +
       , git

     

       13
       13
       +
       , fetchzip ? null

     

       14
       14
       +
       , gtest ? null

     

       15
       15
       +
       , boost ? null

     

       16
       16
       +
       , python3Packages ? null

     

       17
       17
       +
       , buildTests ? false

     

       18
       18
       +
       , buildBenchmarks ? false # Seems to depend on tests

     

       19
       19
       +
       }:

     

       20
       20
       +
       

     

       21
       21
       +
       assert (buildTests || buildBenchmarks) -> fetchzip != null;

     

       22
       22
       +
       assert (buildTests || buildBenchmarks) -> gtest != null;

     

       23
       23
       +
       assert (buildTests || buildBenchmarks) -> boost != null;

     

       24
       24
       +
       assert (buildTests || buildBenchmarks) -> python3Packages != null;

     

       25
       25
       +
       

     

       26
       26
       +
       let

     

       27
       27
       +
         matrices = lib.optionalAttrs (buildTests || buildBenchmarks) import ./deps.nix {

     

       28
       28
       +
           inherit fetchzip;

     

       29
       29
       +
           mirror1 = "https://sparse.tamu.edu/MM";

     

       30
       30
       +
           mirror2 = "https://www.cise.ufl.edu/research/sparse/MM";

     

       31
       31
       +
         };

     

       32
       32
       +
       in stdenv.mkDerivation rec {

     

       33
       33
       +
         pname = "rocsparse";

     

       34
       34
       +
         rocmVersion = "5.3.1";

     

       35
       35
       +
         version = "2.3.2-${rocmVersion}";

     

       36
       36
       +
       

     

       37
       37
       +
         outputs = [

     

       38
       38
       +
           "out"

     

       39
       39
       +
         ] ++ lib.optionals (buildTests || buildBenchmarks) [

     

       40
       40
       +
           "test"

     

       41
       41
       +
         ] ++ lib.optionals buildBenchmarks [

     

       42
       42
       +
           "benchmark"

     

       43
       43
       +
         ];

     

       44
       44
       +
       

     

       45
       45
       +
         src = fetchFromGitHub {

     

       46
       46
       +
           owner = "ROCmSoftwarePlatform";

     

       47
       47
       +
           repo = "rocSPARSE";

     

       48
       48
       +
           rev = "rocm-${rocmVersion}";

     

       49
       49
       +
           hash = "sha256-1069oBrIpZ4M9CAkzoQ9a5j3WlCXErirTbgTUZuT6b0=";

     

       50
       50
       +
         };

     

       51
       51
       +
       

     

       52
       52
       +
         nativeBuildInputs = [

     

       53
       53
       +
           cmake

     

       54
       54
       +
           rocm-cmake

     

       55
       55
       +
           hip

     

       56
       56
       +
           gfortran

     

       57
       57
       +
         ];

     

       58
       58
       +
       

     

       59
       59
       +
         buildInputs = [

     

       60
       60
       +
           rocm-runtime

     

       61
       61
       +
           rocm-device-libs

     

       62
       62
       +
           rocm-comgr

     

       63
       63
       +
           rocprim

     

       64
       64
       +
           git

     

       65
       65
       +
         ] ++ lib.optionals (buildTests || buildBenchmarks) [

     

       66
       66
       +
           gtest

     

       67
       67
       +
           boost

     

       68
       68
       +
           python3Packages.python

     

       69
       69
       +
           python3Packages.pyyaml

     

       70
       70
       +
         ];

     

       71
       71
       +
       

     

       72
       72
       +
         cmakeFlags = [

     

       73
       73
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       74
       74
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       75
       75
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       76
       76
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       77
       77
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       78
       78
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       79
       79
       +
         ] ++ lib.optionals (buildTests || buildBenchmarks) [

     

       80
       80
       +
           "-DBUILD_CLIENTS_TESTS=ON"

     

       81
       81
       +
           "-DCMAKE_MATRICES_DIR=/build/source/matrices"

     

       82
       82
       +
           "-Dpython=python3"

     

       83
       83
       +
         ] ++ lib.optionals buildBenchmarks [

     

       84
       84
       +
           "-DBUILD_CLIENTS_BENCHMARKS=ON"

     

       85
       85
       +
         ];

     

       86
       86
       +
       

     

       87
       87
       +
         # We have to manually generate the matrices

     

       88
       88
       +
         postPatch = lib.optionalString (buildTests || buildBenchmarks) ''

     

       89
       89
       +
           mkdir -p matrices

     

       90
       90
       +
       

     

       91
       91
       +
           ln -s ${matrices.matrix-01}/*.mtx matrices

     

       92
       92
       +
           ln -s ${matrices.matrix-02}/*.mtx matrices

     

       93
       93
       +
           ln -s ${matrices.matrix-03}/*.mtx matrices

     

       94
       94
       +
           ln -s ${matrices.matrix-04}/*.mtx matrices

     

       95
       95
       +
           ln -s ${matrices.matrix-05}/*.mtx matrices

     

       96
       96
       +
           ln -s ${matrices.matrix-06}/*.mtx matrices

     

       97
       97
       +
           ln -s ${matrices.matrix-07}/*.mtx matrices

     

       98
       98
       +
           ln -s ${matrices.matrix-08}/*.mtx matrices

     

       99
       99
       +
           ln -s ${matrices.matrix-09}/*.mtx matrices

     

       100
       100
       +
           ln -s ${matrices.matrix-10}/*.mtx matrices

     

       101
       101
       +
           ln -s ${matrices.matrix-11}/*.mtx matrices

     

       102
       102
       +
           ln -s ${matrices.matrix-12}/*.mtx matrices

     

       103
       103
       +
           ln -s ${matrices.matrix-13}/*.mtx matrices

     

       104
       104
       +
           ln -s ${matrices.matrix-14}/*.mtx matrices

     

       105
       105
       +
           ln -s ${matrices.matrix-15}/*.mtx matrices

     

       106
       106
       +
           ln -s ${matrices.matrix-16}/*.mtx matrices

     

       107
       107
       +
           ln -s ${matrices.matrix-17}/*.mtx matrices

     

       108
       108
       +
           ln -s ${matrices.matrix-18}/*.mtx matrices

     

       109
       109
       +
           ln -s ${matrices.matrix-19}/*.mtx matrices

     

       110
       110
       +
           ln -s ${matrices.matrix-20}/*.mtx matrices

     

       111
       111
       +
           ln -s ${matrices.matrix-21}/*.mtx matrices

     

       112
       112
       +
           ln -s ${matrices.matrix-22}/*.mtx matrices

     

       113
       113
       +
           ln -s ${matrices.matrix-23}/*.mtx matrices

     

       114
       114
       +
           ln -s ${matrices.matrix-24}/*.mtx matrices

     

       115
       115
       +
       

     

       116
       116
       +
           # Not used by the original cmake, causes an error

     

       117
       117
       +
           rm matrices/*_b.mtx

     

       118
       118
       +
       

     

       119
       119
       +
           echo "deps/convert.cpp -> deps/mtx2csr"

     

       120
       120
       +
           hipcc deps/convert.cpp -O3 -o deps/mtx2csr

     

       121
       121
       +
       

     

       122
       122
       +
           for mat in $(ls -1 matrices | cut -d "." -f 1); do

     

       123
       123
       +
             echo "mtx2csr: $mat.mtx -> $mat.csr"

     

       124
       124
       +
             deps/mtx2csr matrices/$mat.mtx matrices/$mat.csr

     

       125
       125
       +
             unlink matrices/$mat.mtx

     

       126
       126
       +
           done

     

       127
       127
       +
         '';

     

       128
       128
       +
       

     

       129
       129
       +
         postInstall = lib.optionalString buildBenchmarks ''

     

       130
       130
       +
           mkdir -p $benchmark/bin

     

       131
       131
       +
           cp -a $out/bin/* $benchmark/bin

     

       132
       132
       +
           rm $benchmark/bin/rocsparse-test

     

       133
       133
       +
         '' + lib.optionalString (buildTests || buildBenchmarks) ''

     

       134
       134
       +
           mkdir -p $test/bin

     

       135
       135
       +
           mv $out/bin/* $test/bin

     

       136
       136
       +
           rm $test/bin/rocsparse-bench || true

     

       137
       137
       +
           mv /build/source/matrices $test

     

       138
       138
       +
           rmdir $out/bin

     

       139
       139
       +
         '';

     

       140
       140
       +
       

     

       141
       141
       +
         meta = with lib; {

     

       142
       142
       +
           description = "ROCm SPARSE implementation";

     

       143
       143
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rocSPARSE";

     

       144
       144
       +
           license = with licenses; [ mit ];

     

       145
       145
       +
           maintainers = with maintainers; [ Madouura ];

     

       146
       146
       +
           broken = rocmVersion != hip.version;

     

       147
       147
       +
         };

     

       148
       148
       +
       }

+222

pkgs/development/libraries/rocsparse/deps.nix

···

       1
       1
       +
       { fetchzip

     

       2
       2
       +
       , mirror1

     

       3
       3
       +
       , mirror2

     

       4
       4
       +
       }:

     

       5
       5
       +
       

     

       6
       6
       +
       {

     

       7
       7
       +
         matrix-01 = fetchzip {

     

       8
       8
       +
           sha256 = "sha256-AHur5ZIDZTFRrO2GV0ieXrffq4KUiGWiZ59pv0fUtEQ=";

     

       9
       9
       +
       

     

       10
       10
       +
           urls = [

     

       11
       11
       +
             "${mirror1}/SNAP/amazon0312.tar.gz"

     

       12
       12
       +
             "${mirror2}/SNAP/amazon0312.tar.gz"

     

       13
       13
       +
           ];

     

       14
       14
       +
         };

     

       15
       15
       +
       

     

       16
       16
       +
         matrix-02 = fetchzip {

     

       17
       17
       +
           sha256 = "sha256-0rSxaN4lQcdaCLsvlgicG70FXUxXeERPiEmQ4MzbRdE=";

     

       18
       18
       +
       

     

       19
       19
       +
           urls = [

     

       20
       20
       +
             "${mirror1}/Muite/Chebyshev4.tar.gz"

     

       21
       21
       +
             "${mirror2}/Muite/Chebyshev4.tar.gz"

     

       22
       22
       +
           ];

     

       23
       23
       +
         };

     

       24
       24
       +
       

     

       25
       25
       +
         matrix-03 = fetchzip {

     

       26
       26
       +
           sha256 = "sha256-hDzDWDUnHEyFedX/tMNq83ZH8uWyM4xtZYUUAD3rizo=";

     

       27
       27
       +
       

     

       28
       28
       +
           urls = [

     

       29
       29
       +
             "${mirror1}/FEMLAB/sme3Dc.tar.gz"

     

       30
       30
       +
             "${mirror2}/FEMLAB/sme3Dc.tar.gz"

     

       31
       31
       +
           ];

     

       32
       32
       +
         };

     

       33
       33
       +
       

     

       34
       34
       +
         matrix-04 = fetchzip {

     

       35
       35
       +
           sha256 = "sha256-GmN2yOt/MoX01rKe05aTyB3ypUP4YbQGOITZ0BqPmC0=";

     

       36
       36
       +
       

     

       37
       37
       +
           urls = [

     

       38
       38
       +
             "${mirror1}/Williams/webbase-1M.tar.gz"

     

       39
       39
       +
             "${mirror2}/Williams/webbase-1M.tar.gz"

     

       40
       40
       +
           ];

     

       41
       41
       +
         };

     

       42
       42
       +
       

     

       43
       43
       +
         matrix-05 = fetchzip {

     

       44
       44
       +
           sha256 = "sha256-gQNjfVyWzNM9RwImJGhkhahRmZz74LzDs1oijL7mI7k=";

     

       45
       45
       +
       

     

       46
       46
       +
           urls = [

     

       47
       47
       +
             "${mirror1}/Williams/mac_econ_fwd500.tar.gz"

     

       48
       48
       +
             "${mirror2}/Williams/mac_econ_fwd500.tar.gz"

     

       49
       49
       +
           ];

     

       50
       50
       +
         };

     

       51
       51
       +
       

     

       52
       52
       +
         matrix-06 = fetchzip {

     

       53
       53
       +
           sha256 = "sha256-87cdZjntNcTuz5BtO59irhcuRbPllWSbhCEX3Td02qc=";

     

       54
       54
       +
       

     

       55
       55
       +
           urls = [

     

       56
       56
       +
             "${mirror1}/Williams/mc2depi.tar.gz"

     

       57
       57
       +
             "${mirror2}/Williams/mc2depi.tar.gz"

     

       58
       58
       +
           ];

     

       59
       59
       +
         };

     

       60
       60
       +
       

     

       61
       61
       +
         matrix-07 = fetchzip {

     

       62
       62
       +
           sha256 = "sha256-WRamuJX3D8Tm+k0q67RjUDG3DeNAxhKiaPkk5afY5eU=";

     

       63
       63
       +
       

     

       64
       64
       +
           urls = [

     

       65
       65
       +
             "${mirror1}/Bova/rma10.tar.gz"

     

       66
       66
       +
             "${mirror2}/Bova/rma10.tar.gz"

     

       67
       67
       +
           ];

     

       68
       68
       +
         };

     

       69
       69
       +
       

     

       70
       70
       +
         matrix-08 = fetchzip {

     

       71
       71
       +
           sha256 = "sha256-5dhkm293Mc3lzakKxHy5W5XIn4Rw+gihVh7gyrjEHXo=";

     

       72
       72
       +
       

     

       73
       73
       +
           urls = [

     

       74
       74
       +
             "${mirror1}/JGD_BIBD/bibd_22_8.tar.gz"

     

       75
       75
       +
             "${mirror2}/JGD_BIBD/bibd_22_8.tar.gz"

     

       76
       76
       +
           ];

     

       77
       77
       +
         };

     

       78
       78
       +
       

     

       79
       79
       +
         matrix-09 = fetchzip {

     

       80
       80
       +
           sha256 = "sha256-czjLWCjXAjZCk5TGYHaEkwSAzQu3TQ3QyB6eNKR4G88=";

     

       81
       81
       +
       

     

       82
       82
       +
           urls = [

     

       83
       83
       +
             "${mirror1}/Hamm/scircuit.tar.gz"

     

       84
       84
       +
             "${mirror2}/Hamm/scircuit.tar.gz"

     

       85
       85
       +
           ];

     

       86
       86
       +
         };

     

       87
       87
       +
       

     

       88
       88
       +
         matrix-10 = fetchzip {

     

       89
       89
       +
           sha256 = "sha256-bYuLnJViAIcIejAkh69/bsNAVIDU4wfTLtD+nmHd6FM=";

     

       90
       90
       +
       

     

       91
       91
       +
           urls = [

     

       92
       92
       +
             "${mirror1}/Sandia/ASIC_320k.tar.gz"

     

       93
       93
       +
             "${mirror2}/Sandia/ASIC_320k.tar.gz"

     

       94
       94
       +
           ];

     

       95
       95
       +
         };

     

       96
       96
       +
       

     

       97
       97
       +
         matrix-11 = fetchzip {

     

       98
       98
       +
           sha256 = "sha256-aDwn8P1khYjo2Agbq5m9ZBInJUxf/knJNvyptt0fak0=";

     

       99
       99
       +
       

     

       100
       100
       +
           urls = [

     

       101
       101
       +
             "${mirror1}/GHS_psdef/bmwcra_1.tar.gz"

     

       102
       102
       +
             "${mirror2}/GHS_psdef/bmwcra_1.tar.gz"

     

       103
       103
       +
           ];

     

       104
       104
       +
         };

     

       105
       105
       +
       

     

       106
       106
       +
         matrix-12 = fetchzip {

     

       107
       107
       +
           sha256 = "sha256-8OJqA/byhlAZd869TPUzZFdsOiwOoRGfKyhM+RMjXoY=";

     

       108
       108
       +
       

     

       109
       109
       +
           urls = [

     

       110
       110
       +
             "${mirror1}/HB/nos1.tar.gz"

     

       111
       111
       +
             "${mirror2}/HB/nos1.tar.gz"

     

       112
       112
       +
           ];

     

       113
       113
       +
         };

     

       114
       114
       +
       

     

       115
       115
       +
         matrix-13 = fetchzip {

     

       116
       116
       +
           sha256 = "sha256-FS0rKqmg+uHwsM/yGfQLBdd7LH/rUrdutkNGBD/Mh1I=";

     

       117
       117
       +
       

     

       118
       118
       +
           urls = [

     

       119
       119
       +
             "${mirror1}/HB/nos2.tar.gz"

     

       120
       120
       +
             "${mirror2}/HB/nos2.tar.gz"

     

       121
       121
       +
           ];

     

       122
       122
       +
         };

     

       123
       123
       +
       

     

       124
       124
       +
         matrix-14 = fetchzip {

     

       125
       125
       +
           sha256 = "sha256-DANnlrNJikrI7Pst9vRedtbuxepyHmCIu2yhltc4Qcs=";

     

       126
       126
       +
       

     

       127
       127
       +
           urls = [

     

       128
       128
       +
             "${mirror1}/HB/nos3.tar.gz"

     

       129
       129
       +
             "${mirror2}/HB/nos3.tar.gz"

     

       130
       130
       +
           ];

     

       131
       131
       +
         };

     

       132
       132
       +
       

     

       133
       133
       +
         matrix-15 = fetchzip {

     

       134
       134
       +
           sha256 = "sha256-21mUgqjWGUfYgiWwSrKh9vH8Vdt3xzcefmqYNYRpxiY=";

     

       135
       135
       +
       

     

       136
       136
       +
           urls = [

     

       137
       137
       +
             "${mirror1}/HB/nos4.tar.gz"

     

       138
       138
       +
             "${mirror2}/HB/nos4.tar.gz"

     

       139
       139
       +
           ];

     

       140
       140
       +
         };

     

       141
       141
       +
       

     

       142
       142
       +
         matrix-16 = fetchzip {

     

       143
       143
       +
           sha256 = "sha256-FOuXvGqBBFNkVS6cexmkluret54hCfCOdK+DOZllE4c=";

     

       144
       144
       +
       

     

       145
       145
       +
           urls = [

     

       146
       146
       +
             "${mirror1}/HB/nos5.tar.gz"

     

       147
       147
       +
             "${mirror2}/HB/nos5.tar.gz"

     

       148
       148
       +
           ];

     

       149
       149
       +
         };

     

       150
       150
       +
       

     

       151
       151
       +
         matrix-17 = fetchzip {

     

       152
       152
       +
           sha256 = "sha256-+7NI1rA/qQxYPpjXKHvAaCZ+LSaAJ4xuJvMRMBEUYxg=";

     

       153
       153
       +
       

     

       154
       154
       +
           urls = [

     

       155
       155
       +
             "${mirror1}/HB/nos6.tar.gz"

     

       156
       156
       +
             "${mirror2}/HB/nos6.tar.gz"

     

       157
       157
       +
           ];

     

       158
       158
       +
         };

     

       159
       159
       +
       

     

       160
       160
       +
         matrix-18 = fetchzip {

     

       161
       161
       +
           sha256 = "sha256-q3NxJjbwGGcFiQ9nhWfUKgZmdVwCfPmgQoqy0AqOsNc=";

     

       162
       162
       +
       

     

       163
       163
       +
           urls = [

     

       164
       164
       +
             "${mirror1}/HB/nos7.tar.gz"

     

       165
       165
       +
             "${mirror2}/HB/nos7.tar.gz"

     

       166
       166
       +
           ];

     

       167
       167
       +
         };

     

       168
       168
       +
       

     

       169
       169
       +
         matrix-19 = fetchzip {

     

       170
       170
       +
           sha256 = "sha256-0GAN6qmVfD+tprIigzuUUUwm5KVhkN9X65wMEvFltDY=";

     

       171
       171
       +
       

     

       172
       172
       +
           urls = [

     

       173
       173
       +
             "${mirror1}/DNVS/shipsec1.tar.gz"

     

       174
       174
       +
             "${mirror2}/DNVS/shipsec1.tar.gz"

     

       175
       175
       +
           ];

     

       176
       176
       +
         };

     

       177
       177
       +
       

     

       178
       178
       +
         matrix-20 = fetchzip {

     

       179
       179
       +
           sha256 = "sha256-f28Du/Urxsiq5NkRmRO10Zz9vvGRjEchquzHzbZpZ7U=";

     

       180
       180
       +
       

     

       181
       181
       +
           urls = [

     

       182
       182
       +
             "${mirror1}/Cote/mplate.tar.gz"

     

       183
       183
       +
             "${mirror2}/Cote/mplate.tar.gz"

     

       184
       184
       +
           ];

     

       185
       185
       +
         };

     

       186
       186
       +
       

     

       187
       187
       +
         matrix-21 = fetchzip {

     

       188
       188
       +
           sha256 = "sha256-O+Wy0NfCU1hVUOfNR1dJpvDHLBwwa301IRJDrQJnhak=";

     

       189
       189
       +
       

     

       190
       190
       +
           urls = [

     

       191
       191
       +
             "${mirror1}/Bai/qc2534.tar.gz"

     

       192
       192
       +
             "${mirror2}/Bai/qc2534.tar.gz"

     

       193
       193
       +
           ];

     

       194
       194
       +
         };

     

       195
       195
       +
       

     

       196
       196
       +
         matrix-22 = fetchzip {

     

       197
       197
       +
           sha256 = "sha256-oxMnt8U5Cf1ILWcBdU6W9jdSMMm+U6bIVl8nm3n3+OA=";

     

       198
       198
       +
       

     

       199
       199
       +
           urls = [

     

       200
       200
       +
             "${mirror1}/Chevron/Chevron2.tar.gz"

     

       201
       201
       +
             "${mirror2}/Chevron/Chevron2.tar.gz"

     

       202
       202
       +
           ];

     

       203
       203
       +
         };

     

       204
       204
       +
       

     

       205
       205
       +
         matrix-23 = fetchzip {

     

       206
       206
       +
           sha256 = "sha256-MFD9BxFI/3IS7yatW121BAI04fbqrXpgYDT5UKjeKcU=";

     

       207
       207
       +
       

     

       208
       208
       +
           urls = [

     

       209
       209
       +
             "${mirror1}/Chevron/Chevron3.tar.gz"

     

       210
       210
       +
             "${mirror2}/Chevron/Chevron3.tar.gz"

     

       211
       211
       +
           ];

     

       212
       212
       +
         };

     

       213
       213
       +
       

     

       214
       214
       +
         matrix-24 = fetchzip {

     

       215
       215
       +
           sha256 = "sha256-ikS8O51pe1nt3BNyhvfvqCbVL0+bg/da9bqGqeBDkTg=";

     

       216
       216
       +
       

     

       217
       217
       +
           urls = [

     

       218
       218
       +
             "${mirror1}/Chevron/Chevron4.tar.gz"

     

       219
       219
       +
             "${mirror2}/Chevron/Chevron4.tar.gz"

     

       220
       220
       +
           ];

     

       221
       221
       +
         };

     

       222
       222
       +
       }

+91

pkgs/development/libraries/rocthrust/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , cmake

     

       5
       5
       +
       , rocm-cmake

     

       6
       6
       +
       , rocm-runtime

     

       7
       7
       +
       , rocm-device-libs

     

       8
       8
       +
       , rocm-comgr

     

       9
       9
       +
       , rocprim

     

       10
       10
       +
       , hip

     

       11
       11
       +
       , gtest ? null

     

       12
       12
       +
       , buildTests ? false

     

       13
       13
       +
       , buildBenchmarks ? false

     

       14
       14
       +
       }:

     

       15
       15
       +
       

     

       16
       16
       +
       assert buildTests -> gtest != null;

     

       17
       17
       +
       

     

       18
       18
       +
       # Doesn't seem to work, thousands of errors compiling with no clear fix

     

       19
       19
       +
       # Is this an upstream issue? We don't seem to be missing dependencies

     

       20
       20
       +
       assert buildTests == false;

     

       21
       21
       +
       assert buildBenchmarks == false;

     

       22
       22
       +
       

     

       23
       23
       +
       stdenv.mkDerivation rec {

     

       24
       24
       +
         pname = "rocthrust";

     

       25
       25
       +
         rocmVersion = "5.3.1";

     

       26
       26
       +
         version = "2.16.0-${rocmVersion}";

     

       27
       27
       +
       

     

       28
       28
       +
         # Comment out these outputs until tests/benchmarks are fixed (upstream?)

     

       29
       29
       +
         # outputs = [

     

       30
       30
       +
         #   "out"

     

       31
       31
       +
         # ] ++ lib.optionals buildTests [

     

       32
       32
       +
         #   "test"

     

       33
       33
       +
         # ] ++ lib.optionals buildBenchmarks [

     

       34
       34
       +
         #   "benchmark"

     

       35
       35
       +
         # ];

     

       36
       36
       +
       

     

       37
       37
       +
         src = fetchFromGitHub {

     

       38
       38
       +
           owner = "ROCmSoftwarePlatform";

     

       39
       39
       +
           repo = "rocThrust";

     

       40
       40
       +
           rev = "rocm-${rocmVersion}";

     

       41
       41
       +
           hash = "sha256-cT0VyEVz86xR6qubAY2ncTxtCRTwXrNTWcFyf3mV+y0=";

     

       42
       42
       +
         };

     

       43
       43
       +
       

     

       44
       44
       +
         nativeBuildInputs = [

     

       45
       45
       +
           cmake

     

       46
       46
       +
           rocm-cmake

     

       47
       47
       +
           rocprim

     

       48
       48
       +
           hip

     

       49
       49
       +
         ];

     

       50
       50
       +
       

     

       51
       51
       +
         buildInputs = [

     

       52
       52
       +
           rocm-runtime

     

       53
       53
       +
           rocm-device-libs

     

       54
       54
       +
           rocm-comgr

     

       55
       55
       +
         ] ++ lib.optionals buildTests [

     

       56
       56
       +
           gtest

     

       57
       57
       +
         ];

     

       58
       58
       +
       

     

       59
       59
       +
         cmakeFlags = [

     

       60
       60
       +
           "-DCMAKE_CXX_COMPILER=hipcc"

     

       61
       61
       +
           "-DHIP_ROOT_DIR=${hip}"

     

       62
       62
       +
           # Manually define CMAKE_INSTALL_<DIR>

     

       63
       63
       +
           # See: https://github.com/NixOS/nixpkgs/pull/197838

     

       64
       64
       +
           "-DCMAKE_INSTALL_BINDIR=bin"

     

       65
       65
       +
           "-DCMAKE_INSTALL_LIBDIR=lib"

     

       66
       66
       +
           "-DCMAKE_INSTALL_INCLUDEDIR=include"

     

       67
       67
       +
         ] ++ lib.optionals buildTests [

     

       68
       68
       +
           "-DBUILD_TEST=ON"

     

       69
       69
       +
         ] ++ lib.optionals buildBenchmarks [

     

       70
       70
       +
           "-DBUILD_BENCHMARKS=ON"

     

       71
       71
       +
         ];

     

       72
       72
       +
       

     

       73
       73
       +
         # Comment out these outputs until tests/benchmarks are fixed (upstream?)

     

       74
       74
       +
         # postInstall = lib.optionalString buildTests ''

     

       75
       75
       +
         #   mkdir -p $test/bin

     

       76
       76
       +
         #   mv $out/bin/test_* $test/bin

     

       77
       77
       +
         # '' + lib.optionalString buildBenchmarks ''

     

       78
       78
       +
         #   mkdir -p $benchmark/bin

     

       79
       79
       +
         #   mv $out/bin/benchmark_* $benchmark/bin

     

       80
       80
       +
         # '' + lib.optionalString (buildTests || buildBenchmarks) ''

     

       81
       81
       +
         #   rmdir $out/bin

     

       82
       82
       +
         # '';

     

       83
       83
       +
       

     

       84
       84
       +
         meta = with lib; {

     

       85
       85
       +
           description = "ROCm parallel algorithm library";

     

       86
       86
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/rocThrust";

     

       87
       87
       +
           license = with licenses; [ asl20 ];

     

       88
       88
       +
           maintainers = with maintainers; [ Madouura ];

     

       89
       89
       +
           broken = rocmVersion != hip.version;

     

       90
       90
       +
         };

     

       91
       91
       +
       }

+34

pkgs/development/libraries/tensile/default.nix

···

       1
       1
       +
       { lib

     

       2
       2
       +
       , stdenv

     

       3
       3
       +
       , fetchFromGitHub

     

       4
       4
       +
       , buildPythonPackage

     

       5
       5
       +
       , pyyaml

     

       6
       6
       +
       , msgpack

     

       7
       7
       +
       , pandas

     

       8
       8
       +
       }:

     

       9
       9
       +
       

     

       10
       10
       +
       buildPythonPackage rec {

     

       11
       11
       +
         pname = "tensile";

     

       12
       12
       +
         rocmVersion = "5.3.1";

     

       13
       13
       +
         version = "4.34.0-${rocmVersion}";

     

       14
       14
       +
       

     

       15
       15
       +
         src = fetchFromGitHub {

     

       16
       16
       +
           owner = "ROCmSoftwarePlatform";

     

       17
       17
       +
           repo = "Tensile";

     

       18
       18
       +
           rev = "rocm-${rocmVersion}";

     

       19
       19
       +
           hash = "sha256-QWt/zzBrZKM8h3MTnbLX4vN3p6cCQvo67U1C2yqAQxw=";

     

       20
       20
       +
         };

     

       21
       21
       +
       

     

       22
       22
       +
         buildInputs = [

     

       23
       23
       +
           pyyaml

     

       24
       24
       +
           msgpack

     

       25
       25
       +
           pandas

     

       26
       26
       +
         ];

     

       27
       27
       +
       

     

       28
       28
       +
         meta = with lib; {

     

       29
       29
       +
           description = "GEMMs and tensor contractions";

     

       30
       30
       +
           homepage = "https://github.com/ROCmSoftwarePlatform/Tensile";

     

       31
       31
       +
           license = with licenses; [ mit ];

     

       32
       32
       +
           maintainers = with maintainers; [ Madouura ];

     

       33
       33
       +
         };

     

       34
       34
       +
       }

+1 -1

pkgs/development/tools/build-managers/rocm-cmake/default.nix

···

       2
       2
        
       

     

       3
       3
        
       stdenv.mkDerivation rec {

     

       4
       4
        
         pname = "rocm-cmake";

     

       5
       5
       -
         version = "5.3.0";

     

       5
       5
       +
         version = "5.3.1";

     

       6
       6
        
       

     

       7
       7
        
         src = fetchFromGitHub {

     

       8
       8
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/development/tools/rocminfo/default.nix

···

       7
       7
        
         # compilers to determine the desired target.

     

       8
       8
        
       , defaultTargets ? []}:

     

       9
       9
        
       stdenv.mkDerivation rec {

     

       10
       10
       -
         version = "5.3.0";

     

       10
       10
       +
         version = "5.3.1";

     

       11
       11
        
         pname = "rocminfo";

     

       12
       12
        
         src = fetchFromGitHub {

     

       13
       13
        
           owner = "RadeonOpenCompute";

+1 -1

pkgs/tools/system/rocm-smi/default.nix

···

       2
       2
        
       

     

       3
       3
        
       stdenv.mkDerivation rec {

     

       4
       4
        
         pname = "rocm-smi";

     

       5
       5
       -
         version = "5.3.0";

     

       5
       5
       +
         version = "5.3.1";

     

       6
       6
        
       

     

       7
       7
        
         src = fetchFromGitHub {

     

       8
       8
        
           owner = "RadeonOpenCompute";

+32

pkgs/top-level/all-packages.nix

···

       14847
       14847
        
       

     

       14848
       14848
        
         rgbds = callPackage ../development/compilers/rgbds { };

     

       14849
       14849
        
       

     

       14850
       14850
       +
         composable_kernel = callPackage ../development/libraries/composable_kernel {

     

       14851
       14851
       +
           inherit (llvmPackages) openmp;

     

       14852
       14852
       +
         };

     

       14853
       14853
       +
       

     

       14854
       14854
       +
         clang-ocl = callPackage ../development/libraries/clang-ocl {

     

       14855
       14855
       +
           inherit (llvmPackages_rocm) clang;

     

       14856
       14856
       +
         };

     

       14857
       14857
       +
       

     

       14850
       14858
        
         rgxg = callPackage ../tools/text/rgxg { };

     

       14851
       14859
        
       

     

       14852
       14860
        
         rocclr = callPackage ../development/libraries/rocclr { };

     
···

       14854
       14862
        
         hip = callPackage ../development/compilers/hip {

     

       14855
       14863
        
           inherit (llvmPackages_rocm) clang llvm;

     

       14856
       14864
        
         };

     

       14865
       14865
       +
       

     

       14866
       14866
       +
         hipcub = callPackage ../development/libraries/hipcub { };

     

       14867
       14867
       +
       

     

       14868
       14868
       +
         hipsparse = callPackage ../development/libraries/hipsparse { };

     

       14869
       14869
       +
       

     

       14870
       14870
       +
         rccl = callPackage ../development/libraries/rccl { };

     

       14857
       14871
        
       

     

       14858
       14872
        
         rocm-cmake = callPackage ../development/tools/build-managers/rocm-cmake { };

     

       14859
       14873
        
       

     
···

       14882
       14896
        
         rocminfo = callPackage ../development/tools/rocminfo { };

     

       14883
       14897
        
       

     

       14884
       14898
        
         rocmlir = callPackage ../development/libraries/rocmlir { };

     

       14899
       14899
       +
       

     

       14900
       14900
       +
         rocprim = callPackage ../development/libraries/rocprim { };

     

       14901
       14901
       +
       

     

       14902
       14902
       +
         rocsparse = callPackage ../development/libraries/rocsparse { };

     

       14903
       14903
       +
       

     

       14904
       14904
       +
         rocfft = callPackage ../development/libraries/rocfft { };

     

       14905
       14905
       +
       

     

       14906
       14906
       +
         tensile = python3Packages.callPackage ../development/libraries/tensile { };

     

       14907
       14907
       +
       

     

       14908
       14908
       +
         rocblas = callPackage ../development/libraries/rocblas {

     

       14909
       14909
       +
           inherit (llvmPackages_rocm) llvm;

     

       14910
       14910
       +
         };

     

       14911
       14911
       +
       

     

       14912
       14912
       +
         miopengemm = callPackage ../development/libraries/miopengemm {

     

       14913
       14913
       +
           inherit (llvmPackages_rocm) clang;

     

       14914
       14914
       +
         };

     

       14915
       14915
       +
       

     

       14916
       14916
       +
         rocthrust = callPackage ../development/libraries/rocthrust { };

     

       14885
       14917
        
       

     

       14886
       14918
        
         rtags = callPackage ../development/tools/rtags {

     

       14887
       14919
        
           inherit (darwin) apple_sdk;