Unverified Commit 38008ffd authored by Martin Weinelt's avatar Martin Weinelt Committed by GitHub
Browse files

Merge pull request #310406 from getchoo/pkgs/accelerate/fix-test 

python3Packages.accelerate: disable problematic tests on aarch64-linux
parents 0838b0da e745fd5a

pkgs/development/python-modules/accelerate/default.nix

+2 −0
Original line number Diff line number Diff line
@@ -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"