···

       
4
       
 
       
  fetchFromGitHub,

     

       
5
       
 
       
  cmake,

     

       
6
       
 
       
  clr,

     

       
7
       
-
       
  gcc,

     

       
8
       
 
       
  python3,

     

       
9
       
 
       
}:

     

       
10
       
 
       


     
···

       
20
       
 
       
  };

     

       
21
       
 
       


     

       
22
       
 
       
  postPatch = ''

     

       
23
       
-
       
    g++ contrib/easy-encryption/cl.cpp -o contrib/easy-encryption/bin/linux/ee64 #replacing prebuilt binary

     

       
24
       
 
       
    substituteInPlace contrib/Orochi/contrib/hipew/src/hipew.cpp --replace-fail '"/opt/rocm/hip/lib/' '"${clr}/lib'

     

       
25
       
 
       
    substituteInPlace hiprt/hiprt_libpath.h --replace-fail '"/opt/rocm/hip/lib/' '"${clr}/lib/'

     

       
26
       
 
       
  '';

     

       
27
       
 
       


     

       
28
       
 
       
  nativeBuildInputs = [

     

       
29
       
-
       
    gcc # required for replacing easy-encryption binary

     

       
30
       
 
       
    cmake

     

       
31
       
 
       
    python3

     

       
32
       
 
       
  ];

     
···

       
36
       
 
       
  ];

     

       
37
       
 
       


     

       
38
       
 
       
  cmakeFlags = [

     

       
39
       
-
       
    "-D CMAKE_BUILD_TYPE=Release"

     

       
40
       
-
       
    "-D BAKE_KERNEL=OFF"

     

       
41
       
-
       
    "-D BAKE_COMPILED_KERNEL=OFF"

     

       
42
       
-
       
    "-D BITCODE=ON"

     

       
43
       
-
       
    "-D PRECOMPILE=ON"

     

       
44
       
-
       
    "-D NO_UNITTEST=ON"

     

       
45
       
-
       
    "-D FORCE_DISABLE_CUDA=ON"

     

       
0
       
 
       

     

       
0
       
 
       

     

       
46
       
 
       
  ];

     

       
47
       
 
       


     

       
48
       
 
       
  postInstall = ''

     
···

       
58
       
 
       
    maintainers = with lib.maintainers; [

     

       
59
       
 
       
      mksafavi

     

       
60
       
 
       
    ];

     

       
0
       
 
       

     

       
61
       
 
       
    platforms = lib.platforms.linux;

     

       
62
       
 
       
  };

     

       
63
       
 
       
})

     

···

       
4
       
 
       
  fetchFromGitHub,

     

       
5
       
 
       
  cmake,

     

       
6
       
 
       
  clr,

     

       
0
       
 
       

     

       
7
       
 
       
  python3,

     

       
8
       
 
       
}:

     

       
9
       
 
       


     
···

       
19
       
 
       
  };

     

       
20
       
 
       


     

       
21
       
 
       
  postPatch = ''

     

       
22
       
+
       
    rm -rf contrib/easy-encrypt # contains prebuilt easy-encrypt binaries, we disable encryption

     

       
23
       
 
       
    substituteInPlace contrib/Orochi/contrib/hipew/src/hipew.cpp --replace-fail '"/opt/rocm/hip/lib/' '"${clr}/lib'

     

       
24
       
 
       
    substituteInPlace hiprt/hiprt_libpath.h --replace-fail '"/opt/rocm/hip/lib/' '"${clr}/lib/'

     

       
25
       
 
       
  '';

     

       
26
       
 
       


     

       
27
       
 
       
  nativeBuildInputs = [

     

       
0
       
 
       

     

       
28
       
 
       
    cmake

     

       
29
       
 
       
    python3

     

       
30
       
 
       
  ];

     
···

       
34
       
 
       
  ];

     

       
35
       
 
       


     

       
36
       
 
       
  cmakeFlags = [

     

       
37
       
+
       
    (lib.cmakeBool "BAKE_KERNEL" false)

     

       
38
       
+
       
    (lib.cmakeBool "BAKE_COMPILED_KERNEL" false)

     

       
39
       
+
       
    (lib.cmakeBool "BITCODE" true)

     

       
40
       
+
       
    (lib.cmakeBool "PRECOMPILE" true)

     

       
41
       
+
       
    # needs accelerator

     

       
42
       
+
       
    (lib.cmakeBool "NO_UNITTEST" true)

     

       
43
       
+
       
    # we have no need to support baking encrypted kernels into object files

     

       
44
       
+
       
    (lib.cmakeBool "NO_ENCRYPT" true)

     

       
45
       
+
       
    (lib.cmakeBool "FORCE_DISABLE_CUDA" true)

     

       
46
       
 
       
  ];

     

       
47
       
 
       


     

       
48
       
 
       
  postInstall = ''

     
···

       
58
       
 
       
    maintainers = with lib.maintainers; [

     

       
59
       
 
       
      mksafavi

     

       
60
       
 
       
    ];

     

       
61
       
+
       
    teams = [ lib.teams.rocm ];

     

       
62
       
 
       
    platforms = lib.platforms.linux;

     

       
63
       
 
       
  };

     

       
64
       
 
       
})