gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.torchao: disable failing tests on aarch64-darwin

Browse Source
This commit is contained in:
Sarah Clark
2025-12-23 12:25:28 -08:00
parent 91f18a8227
commit 4ea3b99591
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torchao/default.nix
+13
View File
@@ -185,6 +185,19 @@ buildPythonPackage rec {
# RuntimeError: quantized engine NoQEngine is not supported
"test_smooth_linear_cpu"
"test_smooth_linear_edge_cases"
# TypeError: Trying to convert Float8_e4m3fn to the MPS backend but it does not have support for that dtype.
"test_subclass_slice_subclass2_shape0_device_mps"
"test_subclass_slice_subclass2_shape1_device_mps"
# torch._inductor.exc.InductorError: KeyError: torch.float8_e4m3fn
"test_optim_default_dtype_bf16_optim_name_AdamFp8_device_mps"
"test_optim_smoke_optim_name_AdamWFp8_bfloat16_device_mps"
"test_optim_smoke_optim_name_AdamWFp8_float32_device_mps"
"test_param_groups_optim_name_AdamFp8_device_mps"
"test_subclass_slice_subclass0_shape0_device_mps"
"test_optim_smoke_optim_name_AdamFp8_bfloat16_device_mps"
"test_optim_smoke_optim_name_AdamFp8_float32_device_mps"
"test_subclass_slice_subclass0_shape1_device_mps"
]
++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [
# Flaky: [gw0] node down: keyboard-interrupt