rocmPackages.ck4inductor: don't depend on rocm-merged-llvm

This commit is contained in:
Luna Nova
2025-09-24 07:13:25 -07:00
parent 645a298fb3
commit 766dabeb38
2 changed files with 3 additions and 3 deletions

View File

@@ -3,9 +3,9 @@
python,
composable_kernel,
lib,
rocm-toolchain,
setuptools,
setuptools-scm,
rocm-merged-llvm,
}:
buildPythonPackage {
pyproject = true;
@@ -25,7 +25,7 @@ buildPythonPackage {
propagatedBuildInputs = [
# At runtime will fail to compile anything with ck4inductor without this
# can't easily use in checks phase because most of the compiler machinery is in torch
rocm-merged-llvm
rocm-toolchain
];
checkPhase = ''
if [ ! -d "$out/${python.sitePackages}/ck4inductor" ]; then

View File

@@ -177,7 +177,7 @@ let
ck4inductor = pyPackages.callPackage ./composable_kernel/ck4inductor.nix {
inherit (self) composable_kernel;
inherit (llvm) rocm-merged-llvm;
inherit (llvm) rocmcxx;
};
half = self.callPackage ./half { };