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

pkgs/development/rocm-modules/6/tensile/default.nix

+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" ];