python3Packages.accelerate: skip failing test on aarch64-linux (#479141)
This commit is contained in:
@@ -31,7 +31,7 @@
|
||||
writableTmpDirAsHomeHook,
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
buildPythonPackage (finalAttrs: {
|
||||
pname = "accelerate";
|
||||
version = "1.12.0";
|
||||
pyproject = true;
|
||||
@@ -39,7 +39,7 @@ buildPythonPackage rec {
|
||||
src = fetchFromGitHub {
|
||||
owner = "huggingface";
|
||||
repo = "accelerate";
|
||||
tag = "v${version}";
|
||||
tag = "v${finalAttrs.version}";
|
||||
hash = "sha256-PwwaQSLOm+8Hd3trM1P+jRhYyoWM3QxOe5XT99haEmg=";
|
||||
};
|
||||
|
||||
@@ -108,6 +108,13 @@ buildPythonPackage rec {
|
||||
# TypeError: unsupported operand type(s) for /: 'NoneType' and 'int' (it seems cpuinfo doesn't work here)
|
||||
"test_mpi_multicpu_config_cmd"
|
||||
]
|
||||
++
|
||||
lib.optionals
|
||||
(stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64 && (pythonAtLeast "3.14"))
|
||||
[
|
||||
# RuntimeError: There is no current event loop in thread 'MainThread'
|
||||
"test_accelerate_test"
|
||||
]
|
||||
++ lib.optionals (!config.cudaSupport) [
|
||||
# requires ptxas from cudatoolkit, which is unfree
|
||||
"test_dynamo_extract_model"
|
||||
@@ -187,9 +194,9 @@ buildPythonPackage rec {
|
||||
meta = {
|
||||
homepage = "https://huggingface.co/docs/accelerate";
|
||||
description = "Simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision";
|
||||
changelog = "https://github.com/huggingface/accelerate/releases/tag/${src.tag}";
|
||||
changelog = "https://github.com/huggingface/accelerate/releases/tag/${finalAttrs.src.tag}";
|
||||
license = lib.licenses.asl20;
|
||||
maintainers = with lib.maintainers; [ bcdarwin ];
|
||||
mainProgram = "accelerate";
|
||||
};
|
||||
}
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user