From 4ea3b99591a053ef0dcb8e51234c74a4a8416d9c Mon Sep 17 00:00:00 2001 From: Sarah Clark Date: Tue, 16 Dec 2025 19:28:09 -0800 Subject: [PATCH] python3Packages.torchao: disable failing tests on aarch64-darwin --- pkgs/development/python-modules/torchao/default.nix | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/pkgs/development/python-modules/torchao/default.nix b/pkgs/development/python-modules/torchao/default.nix index 48a8f34a9817..c2cf15e406f2 100644 --- a/pkgs/development/python-modules/torchao/default.nix +++ b/pkgs/development/python-modules/torchao/default.nix @@ -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