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
fc590fdd05.
Co-authored-by: Mauricio Collares <mauricio@collares.org>
This commit is contained in:
committed by
Mauricio Collares
parent
c056e45ecb
commit
74858c1939
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user