archives/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-03-07 23:04:00 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

python3Packages.flashinfer: 0.6.1 -> 0.6.4

Browse source 
Diff: https://github.com/flashinfer-ai/flashinfer/compare/v0.6.1...v0.6.4
This commit is contained in:
Gaetan Lepage 2026-02-28 10:30:34 +00:00
parent 3f023e4ceb
commit f5c6e65f0f
1 changed files with 10 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

25
pkgs/development/python-modules/flashinfer/default.nix
View file

@ -9,9 +9,9 @@
config,
buildPythonPackage,
fetchFromGitHub,
fetchpatch2,
# build-system
apache-tvm-ffi,
setuptools,
# nativeBuildInputs
@ -29,30 +29,24 @@
tqdm,
}:
buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
pname = "flashinfer";
version = "0.6.1";
version = "0.6.4";
pyproject = true;
src = fetchFromGitHub {
owner = "flashinfer-ai";
repo = "flashinfer";
tag = "v${version}";
tag = "v${finalAttrs.version}";
fetchSubmodules = true;
hash = "sha256-NRjas11VvvCY6MZiZaYtxG5MXEaFqfbhJxflUT/uraE=";
hash = "sha256-Hq3oTeEJHRvXwThI8vc06E3Ot/FnPP0sZUfze3ISa2o=";
};
patches = [
# TODO: remove patch with update to v0.5.2+
# Switch pynvml to nvidia-ml-py
(fetchpatch2 {
url = "https://github.com/flashinfer-ai/flashinfer/commit/a42f99255d68d1a54b689bd4985339c6b44963a6.patch?full_index=1";
hash = "sha256-3XJFcdQeZ/c5fwiQvd95z4p9BzTn8pjle21WzeBxUgk=";
})
build-system = [
apache-tvm-ffi
setuptools
];
build-system = [ setuptools ];
nativeBuildInputs = [
cmake
ninja
@ -91,6 +85,7 @@ buildPythonPackage rec {
pythonRemoveDeps = [
"nvidia-cudnn-frontend"
"nvidia-cutlass-dsl"
];
dependencies = [
click
@ -119,4 +114,4 @@ buildPythonPackage rec {
daniel-fahey
];
};
}
})