From 43cb837c846edd34c334912a26fa08ad10266168 Mon Sep 17 00:00:00 2001 From: Vivek Khandelwal Date: Tue, 2 Apr 2024 10:21:58 +0000 Subject: [PATCH] build: manually update PyTorch version Set PyTorch and TorchVision version to nightly release 2024-04-01. Signed-Off By: Vivek Khandelwal --- .../Torch/Transforms/AbstractInterpLibrary.cpp | 18 ++++++++++++++++++ pytorch-hash.txt | 2 +- pytorch-requirements.txt | 2 +- torchvision-requirements.txt | 2 +- 4 files changed, 21 insertions(+), 3 deletions(-) diff --git a/lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp b/lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp index 38df3af98f675..444fdca684ee6 100644 --- a/lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp +++ b/lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp @@ -5868,6 +5868,24 @@ StringRef mlir::torch::Torch::getAbstractInterpLibrary() { " %3 = torch.prim.TupleConstruct %1, %0, %0 : !torch.list, !torch.list, !torch.list -> !torch.tuple, list, list>\n" " return %3 : !torch.tuple, list, list>\n" " }\n" +" func.func @__torch__.torch.jit._shape_functions._batch_norm_with_update(%arg0: !torch.list, %arg1: !torch.optional>, %arg2: !torch.optional>, %arg3: !torch.optional>, %arg4: !torch.optional>) -> !torch.tuple, list, list, list> {\n" +" %int0 = torch.constant.int 0\n" +" %true = torch.constant.bool true\n" +" %int1 = torch.constant.int 1\n" +" %0 = torch.aten.__getitem__.t %arg0, %int1 : !torch.list, !torch.int -> !torch.int\n" +" %1 = torch.prim.ListConstruct %0 : (!torch.int) -> !torch.list\n" +" %2 = torch.prim.ListConstruct : () -> !torch.list\n" +" %3 = torch.aten.len.t %arg0 : !torch.list -> !torch.int\n" +" torch.prim.Loop %3, %true, init() {\n" +" ^bb0(%arg5: !torch.int):\n" +" %6 = torch.aten.__getitem__.t %arg0, %arg5 : !torch.list, !torch.int -> !torch.int\n" +" %7 = torch.aten.append.t %2, %6 : !torch.list, !torch.int -> !torch.list\n" +" torch.prim.Loop.condition %true, iter()\n" +" } : (!torch.int, !torch.bool) -> ()\n" +" %4 = torch.prim.ListConstruct %int0 : (!torch.int) -> !torch.list\n" +" %5 = torch.prim.TupleConstruct %2, %1, %1, %4 : !torch.list, !torch.list, !torch.list, !torch.list -> !torch.tuple, list, list, list>\n" +" return %5 : !torch.tuple, list, list, list>\n" +" }\n" " func.func @__torch__.torch.jit._shape_functions.cross_entropy_loss(%arg0: !torch.list, %arg1: !torch.list, %arg2: !torch.optional>, %arg3: !torch.int, %arg4: !torch.int, %arg5: !torch.float) -> !torch.list {\n" " %int-1 = torch.constant.int -1\n" " %true = torch.constant.bool true\n" diff --git a/pytorch-hash.txt b/pytorch-hash.txt index 54e47e2f12245..09efb313d6cf6 100644 --- a/pytorch-hash.txt +++ b/pytorch-hash.txt @@ -1 +1 @@ -c823f7a4de48a06d5d57bb92556f772eea1aa83c +e1c8416590b718ab97e06089e57b650ae3909711 diff --git a/pytorch-requirements.txt b/pytorch-requirements.txt index 027c5377dd0d5..f16081073032c 100644 --- a/pytorch-requirements.txt +++ b/pytorch-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre -torch==2.4.0.dev20240318 +torch==2.4.0.dev20240401 diff --git a/torchvision-requirements.txt b/torchvision-requirements.txt index 4110fce6bf8b8..95c189db532c0 100644 --- a/torchvision-requirements.txt +++ b/torchvision-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre -torchvision==0.18.0.dev20240318 +torchvision==0.18.0.dev20240401