python3Packages.compressed-tensors: 0.14.0.1 -> 0.15.0.1
Diff: https://github.com/neuralmagic/compressed-tensors/compare/0.14.0.1...0.15.0.1 Changelog: https://github.com/neuralmagic/compressed-tensors/releases/tag/0.15.0.1
This commit is contained in:
@@ -3,7 +3,6 @@
|
||||
stdenv,
|
||||
buildPythonPackage,
|
||||
fetchFromGitHub,
|
||||
pythonAtLeast,
|
||||
|
||||
# build-system
|
||||
setuptools,
|
||||
@@ -25,7 +24,7 @@
|
||||
|
||||
buildPythonPackage (finalAttrs: {
|
||||
pname = "compressed-tensors";
|
||||
version = "0.14.0.1";
|
||||
version = "0.15.0.1";
|
||||
pyproject = true;
|
||||
|
||||
# Release on PyPI is missing the `utils` directory, which `setup.py` wants to import
|
||||
@@ -33,7 +32,7 @@ buildPythonPackage (finalAttrs: {
|
||||
owner = "neuralmagic";
|
||||
repo = "compressed-tensors";
|
||||
tag = finalAttrs.version;
|
||||
hash = "sha256-bncglC6qRE4D6rjoNOXRKh+7ZMSTc9zc3NVzA+NaXuE=";
|
||||
hash = "sha256-iiYo3Vne2CYlj+wMHxQFTTU7gb8oNwPtCe873nX5KgA=";
|
||||
};
|
||||
|
||||
postPatch = ''
|
||||
@@ -46,7 +45,9 @@ buildPythonPackage (finalAttrs: {
|
||||
setuptools-scm
|
||||
];
|
||||
|
||||
buildInputs = lib.optional stdenv.cc.isClang llvmPackages.openmp;
|
||||
buildInputs = lib.optionals stdenv.cc.isClang [
|
||||
llvmPackages.openmp
|
||||
];
|
||||
|
||||
dependencies = [
|
||||
frozendict
|
||||
@@ -56,10 +57,6 @@ buildPythonPackage (finalAttrs: {
|
||||
transformers
|
||||
];
|
||||
|
||||
pythonRelaxDeps = [ "transformers" ];
|
||||
|
||||
doCheck = true;
|
||||
|
||||
pythonImportsCheck = [ "compressed_tensors" ];
|
||||
|
||||
nativeCheckInputs = [
|
||||
@@ -89,6 +86,11 @@ buildPythonPackage (finalAttrs: {
|
||||
"test_set_item"
|
||||
"test_set_item_buffers"
|
||||
"test_update_offload_parameter_with_grad"
|
||||
]
|
||||
++ lib.optionals stdenv.hostPlatform.isDarwin [
|
||||
# torch._inductor.exc.InductorError: ImportError: dlopen(/nix/var/nix/builds/nix-25002-542173852/torchinductor__nixbld1/xo/cxovsevcfanmw7lgoddbnyhoxes3nzlu7ecugxedaq2zr4f6b2qh.main.so, 0x0002):
|
||||
# symbol not found in flat namespace '___kmpc_barrier'
|
||||
"test_compress_decompress_module"
|
||||
];
|
||||
|
||||
disabledTestPaths = [
|
||||
|
||||
Reference in New Issue
Block a user