Unverified Commit 6ba71cbc authored by Yt's avatar Yt Committed by GitHub
Browse files

python3Packages.torchtune: skip failing tests on x86_64-linux (#482990)

parents 1260a929 ed6b0a3c

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

+10 −0
Original line number Diff line number Diff line 
{

  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 = {

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

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