diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py index 5c613eae0..ced556cf8 100644 --- a/projects/pt1/e2e_testing/xfail_sets.py +++ b/projects/pt1/e2e_testing/xfail_sets.py @@ -505,6 +505,8 @@ FX_IMPORTER_XFAIL_SET = { "ViewCollapseDynamicWithAtenSizeIntModule_basic", "ViewSizeFromOtherTensor_basic", "WeightNormInterfaceModule_basic", + "ScaledDotProductAttentionDifferentModule_basic", + "ScaledDotProductAttentionSameModule_basic", } FX_IMPORTER_CRASHING_SET = LINALG_CRASHING_SET | { @@ -2105,7 +2107,6 @@ MAKE_FX_TOSA_PASS_SET = ( "ViewSizeDimLedAndFollowedByCollapsedOnesModule_basic", "ViewSizeDimLedByCollapsedOnesModule_basic", "ViewSizeFromOtherTensor_basic", - "ScaledDotProductAttentionDifferentModule_basic", "RenormModuleFloat32NegativeDim_basic", "RenormModuleFloat32_basic", } @@ -2148,6 +2149,11 @@ MAKE_FX_TOSA_PASS_SET = ( "AdaptiveAvgPool2dFixedKernelStrideSizeStaticModule_basic", } +if torch_version_for_comparison() < version.parse("2.5.0.dev"): + MAKE_FX_TOSA_PASS_SET = MAKE_FX_TOSA_PASS_SET | { + "ScaledDotProductAttentionDifferentModule_basic", + } + LTC_CRASHING_SET = { # TODO: update test to move all inputs to the lazy device. Otherwise test fails with: # Check failed: lazy_tensor Input tensor is not a lazy tensor: CPUBoolType. diff --git a/pytorch-hash.txt b/pytorch-hash.txt index 95a62d316..11cae2da8 100644 --- a/pytorch-hash.txt +++ b/pytorch-hash.txt @@ -1 +1 @@ -748db193d71a1c29471a87c7841da6a5a0a0dbae +aa1fc68d51488dab6cf353464ea320e2a0db18f8 diff --git a/pytorch-requirements.txt b/pytorch-requirements.txt index 75b0983e9..cae8c406e 100644 --- a/pytorch-requirements.txt +++ b/pytorch-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torch/ --pre -torch==2.5.0.dev20240818 +torch==2.5.0.dev20240825 diff --git a/torchvision-requirements.txt b/torchvision-requirements.txt index 5b5890871..c0acddf9a 100644 --- a/torchvision-requirements.txt +++ b/torchvision-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torchvision/ --pre -torchvision==0.20.0.dev20240818 +torchvision==0.20.0.dev20240825