646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
44 lines
1.2 KiB
Nix
44 lines
1.2 KiB
Nix
{
|
|
buildPythonPackage,
|
|
python,
|
|
composable_kernel,
|
|
lib,
|
|
setuptools,
|
|
setuptools-scm,
|
|
rocm-merged-llvm,
|
|
}:
|
|
buildPythonPackage {
|
|
pyproject = true;
|
|
pname = "ck4inductor";
|
|
build-system = [
|
|
setuptools
|
|
setuptools-scm
|
|
];
|
|
version = "6.4.3";
|
|
inherit (composable_kernel) src;
|
|
pythonImportsCheck = [
|
|
"ck4inductor"
|
|
"ck4inductor.universal_gemm.gen_instances"
|
|
"ck4inductor.universal_gemm.gen_instances"
|
|
"ck4inductor.universal_gemm.op"
|
|
];
|
|
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
|
|
];
|
|
checkPhase = ''
|
|
if [ ! -d "$out/${python.sitePackages}/ck4inductor" ]; then
|
|
echo "ck4inductor isn't at the expected location in $out/${python.sitePackages}/ck4inductor"
|
|
exit 1
|
|
fi
|
|
'';
|
|
meta = with lib; {
|
|
description = "Pytorch inductor backend which uses composable_kernel universal GEMM implementations";
|
|
homepage = "https://github.com/ROCm/composable_kernel";
|
|
license = with licenses; [ mit ];
|
|
teams = [ teams.rocm ];
|
|
platforms = platforms.linux;
|
|
};
|
|
}
|