mirror of https://github.com/llvm/torch-mlir
build: manually update PyTorch version (#3666)
Set PyTorch and TorchVision version to nightly release 2024-08-25. Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>pull/3675/head
parent
6eba5bc9ee
commit
b92e61832f
|
@ -505,6 +505,8 @@ FX_IMPORTER_XFAIL_SET = {
|
||||||
"ViewCollapseDynamicWithAtenSizeIntModule_basic",
|
"ViewCollapseDynamicWithAtenSizeIntModule_basic",
|
||||||
"ViewSizeFromOtherTensor_basic",
|
"ViewSizeFromOtherTensor_basic",
|
||||||
"WeightNormInterfaceModule_basic",
|
"WeightNormInterfaceModule_basic",
|
||||||
|
"ScaledDotProductAttentionDifferentModule_basic",
|
||||||
|
"ScaledDotProductAttentionSameModule_basic",
|
||||||
}
|
}
|
||||||
|
|
||||||
FX_IMPORTER_CRASHING_SET = LINALG_CRASHING_SET | {
|
FX_IMPORTER_CRASHING_SET = LINALG_CRASHING_SET | {
|
||||||
|
@ -2105,7 +2107,6 @@ MAKE_FX_TOSA_PASS_SET = (
|
||||||
"ViewSizeDimLedAndFollowedByCollapsedOnesModule_basic",
|
"ViewSizeDimLedAndFollowedByCollapsedOnesModule_basic",
|
||||||
"ViewSizeDimLedByCollapsedOnesModule_basic",
|
"ViewSizeDimLedByCollapsedOnesModule_basic",
|
||||||
"ViewSizeFromOtherTensor_basic",
|
"ViewSizeFromOtherTensor_basic",
|
||||||
"ScaledDotProductAttentionDifferentModule_basic",
|
|
||||||
"RenormModuleFloat32NegativeDim_basic",
|
"RenormModuleFloat32NegativeDim_basic",
|
||||||
"RenormModuleFloat32_basic",
|
"RenormModuleFloat32_basic",
|
||||||
}
|
}
|
||||||
|
@ -2148,6 +2149,11 @@ MAKE_FX_TOSA_PASS_SET = (
|
||||||
"AdaptiveAvgPool2dFixedKernelStrideSizeStaticModule_basic",
|
"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 = {
|
LTC_CRASHING_SET = {
|
||||||
# TODO: update test to move all inputs to the lazy device. Otherwise test fails with:
|
# 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.
|
# Check failed: lazy_tensor Input tensor is not a lazy tensor: CPUBoolType.
|
||||||
|
|
|
@ -1 +1 @@
|
||||||
748db193d71a1c29471a87c7841da6a5a0a0dbae
|
aa1fc68d51488dab6cf353464ea320e2a0db18f8
|
||||||
|
|
|
@ -1,3 +1,3 @@
|
||||||
-f https://download.pytorch.org/whl/nightly/cpu/torch/
|
-f https://download.pytorch.org/whl/nightly/cpu/torch/
|
||||||
--pre
|
--pre
|
||||||
torch==2.5.0.dev20240818
|
torch==2.5.0.dev20240825
|
||||||
|
|
|
@ -1,3 +1,3 @@
|
||||||
-f https://download.pytorch.org/whl/nightly/cpu/torchvision/
|
-f https://download.pytorch.org/whl/nightly/cpu/torchvision/
|
||||||
--pre
|
--pre
|
||||||
torchvision==0.20.0.dev20240818
|
torchvision==0.20.0.dev20240825
|
||||||
|
|
Loading…
Reference in New Issue