Unverified Commit 05350dbd authored by Luna Nova's avatar Luna Nova Committed by GitHub
Browse files

rocmPackages.composable_kernel: use jemalloc at build time to speed up build by ≈7% (#508689)

parents f190b34c 51a23844

pkgs/development/rocm-modules/composable_kernel/base.nix

+4 −0
Original line number Diff line number Diff line
@@ -11,6 +11,7 @@ 
  hipify,

  gitMinimal,

  gtest,

  jemalloc,

  zstd,

  buildTests ? false,

  buildExamples ? false,
@@ -87,6 +88,9 @@ stdenv.mkDerivation (finalAttrs: { 
  strictDeps = true;

  enableParallelBuilding = true;

  env.ROCM_PATH = clr;

  # Speed up build by ~7% with jemalloc (template torture test workload means allocation heavy clang invocations)

  env.LD_PRELOAD = "${jemalloc}/lib/libjemalloc.so";

  env.MALLOC_CONF = "background_thread:true,metadata_thp:auto,dirty_decay_ms:10000,muzzy_decay_ms:10000";



  cmakeFlags = [

    (lib.cmakeBool "MIOPEN_REQ_LIBS_ONLY" miOpenReqLibsOnly)