Commit 74858c19 authored by Martin Schwaighofer's avatar Martin Schwaighofer Committed by Mauricio Collares
Browse files

rocmPackages.tensile: use unwrapped cc as assembler



Tensile calls rocm clang with parameters such as "-x assembler
-target amdgcn-amd-amdhsa". Using a wrapped compiler with "-target"
leads to warnings introduced in
fc590fdd.

Co-authored-by: default avatarMauricio Collares <mauricio@collares.org>
parent c056e45e
Loading
Loading
Loading
Loading
+7 −0
Original line number Diff line number Diff line
@@ -13,6 +13,7 @@
  joblib,
  filelock,
  rocminfo,
  writeText,
}:

buildPythonPackage rec {
@@ -61,6 +62,12 @@ buildPythonPackage rec {
    ROCM_PATH = rocminfo;
  };

  # TODO: remove this workaround once https://github.com/NixOS/nixpkgs/pull/323869
  # does not cause issues anymore, or at least replace it with a better workaround
  setupHook = writeText "setup-hook" ''
    export TENSILE_ROCM_ASSEMBLER_PATH="${stdenv.cc.cc}/bin/clang++";
  '';

  pythonImportsCheck = [ "Tensile" ];

  passthru.updateScript = rocmUpdateScript {