Unverified Commit d6adc4b1 authored by Gavin Zhao's avatar Gavin Zhao
Browse files

rocmPackages: extend tensile ISA compatibility



Signed-off-by: default avatarGavin Zhao <git@gzgz.dev>
parent f1367529
Loading
Loading
Loading
Loading
+12 −3
Original line number Diff line number Diff line
{ lib
, stdenv
, fetchFromGitHub
, fetchpatch
, rocmUpdateScript
, buildPythonPackage
, pytestCheckHook
@@ -34,6 +35,14 @@ buildPythonPackage rec {
    joblib
  ];

  patches = [
    (fetchpatch {
      name = "Extend-Tensile-HIP-ISA-compatibility.patch";
      url = "https://github.com/GZGavinZhao/Tensile/commit/855cb15839849addb0816a6dde45772034a3e41f.patch";
      hash = "sha256-d+fVf/vz+sxGqJ96vuxe0jRMgbC5K6j5FQ5SJ1e3Sl8=";
    })
  ];

  doCheck = false; # Too many errors, not sure how to set this up properly

  nativeCheckInputs = [
@@ -42,9 +51,9 @@ buildPythonPackage rec {
    rocminfo
  ];

  preCheck = ''
    export ROCM_PATH=${rocminfo}
  '';
  env = {
    ROCM_PATH = rocminfo;
  };

  pythonImportsCheck = [ "Tensile" ];