mirror of https://github.com/llvm/torch-mlir
build: manually update PyTorch version (#3291)
Set PyTorch and TorchVision version to nightly release 2024-05-05. Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>pull/3328/head
parent
7c289d9522
commit
10db310460
|
@ -2660,12 +2660,6 @@ ONNX_XFAIL_SET = {
|
|||
"ReduceMinAlongDimUnsignedInt_basic",
|
||||
}
|
||||
|
||||
if torch_version_for_comparison() >= version.parse("2.4.0.dev"):
|
||||
ONNX_XFAIL_SET = ONNX_XFAIL_SET | {
|
||||
# ERROR: Found dtype (torch.float64) but expected (torch.float32)
|
||||
"ReduceL1NormWithDTypeModule_basic",
|
||||
}
|
||||
|
||||
if torch_version_for_comparison() < version.parse("2.3.0.dev"):
|
||||
ONNX_XFAIL_SET = ONNX_XFAIL_SET | {
|
||||
# ERROR: shape (torch.Size([6, 4, 5])) is not equal to golden shape (torch.Size([120]))
|
||||
|
|
|
@ -1 +1 @@
|
|||
34ade3521ca41f20af3469bba276c2b0499c3892
|
||||
1b7523fbe9d0a0c81930673f4374c6e69fa293b6
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
||||
--pre
|
||||
torch==2.4.0.dev20240428
|
||||
torch==2.4.0.dev20240505
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
||||
--pre
|
||||
torchvision==0.19.0.dev20240428
|
||||
torchvision==0.19.0.dev20240505
|
||||
|
|
Loading…
Reference in New Issue