pkgs/development/libraries/rocmlir/default.nix

   1 { lib
   2 , stdenv
   3 , fetchFromGitHub
   4 , cmake
   5 , hip
   6 , python3
   7 }:
   8
   9 stdenv.mkDerivation rec {
  10   pname = "rocmlir";
  11   rocmVersion = "5.3.1";
  12   # For some reason they didn't add a tag for 5.3.1, should be compatible, change to rocmVersion later
  13   version = "5.3.0";
  14
  15   src = fetchFromGitHub {
  16     owner = "ROCmSoftwarePlatform";
  17     repo = "rocMLIR";
  18     rev = "rocm-${version}"; # change to rocmVersion later
  19     hash = "sha256-s/5gAH5vh2tgATZemPP66juQFDg8BR2sipzX2Q6pOOQ=";
  20   };
  21
  22   nativeBuildInputs = [
  23     cmake
  24     hip
  25   ];
  26
  27   buildInputs = [
  28     python3
  29   ];
  30
  31   cmakeFlags = [
  32     "-DBUILD_FAT_LIBMLIRMIOPEN=ON"
  33   ];
  34
  35   meta = with lib; {
  36     description = "MLIR-based convolution and GEMM kernel generator";
  37     homepage = "https://github.com/ROCmSoftwarePlatform/rocMLIR";
  38     license = with licenses; [ asl20 ];
  39     maintainers = with maintainers; [ Madouura ];
  40     broken = rocmVersion != hip.version;
  41   };
  42 }