nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-10-03 13:30:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.deep-gemm: fix build (#548157)

Browse Source
This commit is contained in:
kirillrdy
2026-08-01 22:30:58 +00:00
committed by GitHub
parent 62080c28fa dcd3a3c1e8
commit d03efe5083
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkgs/development/python-modules/deep-gemm/default.nix
View File

@@ -39,6 +39,7 @@ buildPythonPackage.override { inherit (torch) stdenv; } (finalAttrs: {
pname = "deep-gemm";
version = "2.1.1.post3";
pyproject = true;
__structuredAttrs = true;
src = fetchFromGitHub {
owner = "deepseek-ai";
@@ -68,7 +69,20 @@ buildPythonPackage.override { inherit (torch) stdenv; } (finalAttrs: {
})
];
env = optionalAttrs cudaSupport {
# Upstream hardcodes `__version__ = '2.1.1'` in `deep_gemm/__init__.py`, which does not match the
# release tag and fails the metadata check.
postPatch = ''
substituteInPlace deep_gemm/__init__.py \
--replace-fail \
"__version__ = '2.1.1'" \
"__version__ = '${finalAttrs.version}'"
'';
env = {
# Do not append a `+local` revision to the version (no git repository available)
DG_USE_LOCAL_VERSION = "0";
}
// optionalAttrs cudaSupport {
CUDA_HOME = (getBin cudaPackages.cuda_nvcc).outPath;
LDFLAGS = toString [