diff --git a/pkgs/development/python-modules/executorch/default.nix b/pkgs/development/python-modules/executorch/default.nix index 95d8a94df9504..bc2daeadad6b6 100644 --- a/pkgs/development/python-modules/executorch/default.nix +++ b/pkgs/development/python-modules/executorch/default.nix @@ -1,5 +1,6 @@ { lib, + stdenv, pkgs, buildPythonPackage, fetchFromGitHub, @@ -198,6 +199,15 @@ buildPythonPackage (finalAttrs: { "test_resnet18_export_to_executorch" "test_resnet50_export_to_executorch" "test_vit_export_to_executorch" + ] + ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isx86_64) [ + # RuntimeError: Error in dlopen: + # /tmp/AP8CBk/vision_encoder/data/aotinductor/model/chm6ca425mbz7jdmmwcbnyrm6x6tedrfmaoyr4vee625vxhjibbt.wrapper.so: + # cannot enable executable stack as shared object requires: Invalid argument + "TestImageTransform" + "test_flamingo_vision_encoder" + "test_llama3_2_text_decoder_aoti" + "test_tile_positional_embedding_aoti" ]; meta = { diff --git a/pkgs/development/python-modules/torchtune/default.nix b/pkgs/development/python-modules/torchtune/default.nix index ce757924e9841..61d4a5e1eefa2 100644 --- a/pkgs/development/python-modules/torchtune/default.nix +++ b/pkgs/development/python-modules/torchtune/default.nix @@ -111,6 +111,14 @@ buildPythonPackage (finalAttrs: { "test_forward_with_curr_pos" "test_forward_with_packed_pos" ] + ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isx86_64) [ + # RuntimeError: Error in dlopen: + # /tmp/yae2xK/mha/data/aotinductor/model/ckk2zlroqn6hgq5vvpy7bcjikztqmwqkek3njxe2gvvwp244hjny.wrapper.so: + # cannot enable executable stack as shared object requires: Invalid argument + "test_attention_aoti" + "test_tile_positional_embedding_aoti" + "test_tiled_token_positional_embedding_aoti" + ] ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ # Fatal Python error: Segmentation fault "test_forward_gqa"