python3Packages.accelerate: disable problematic tests on aarch64-linux

2024-05-09 14:46:34 -04:00
parent ac34158a82
commit e745fd5a47
1 changed files with 2 additions and 0 deletions
@@ -81,6 +81,8 @@ buildPythonPackage rec {
  ] ++ lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
    # usual aarch64-linux RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly
    "CheckpointTest"
+    # TypeError: unsupported operand type(s) for /: 'NoneType' and 'int' (it seems cpuinfo doesn't work here)
+    "test_mpi_multicpu_config_cmd"
  ] ++ lib.optionals (!config.cudaSupport) [
    # requires ptxas from cudatoolkit, which is unfree
    "test_dynamo_extract_model"