build: manually update PyTorch version

Set PyTorch and TorchVision version to nightly release 2023-08-10.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
pull/2391/head
Vivek Khandelwal 2023-08-10 13:20:51 +00:00
parent 3854e9e5c3
commit f0a8f273f7
5 changed files with 5 additions and 5 deletions

View File

@ -10449,7 +10449,7 @@ StringRef mlir::torch::Torch::getAbstractInterpLibrary() {
" func.func @\"__torch_mlir_dtype_fn.aten.to.dtype\"(%arg0: !torch.tuple<int, int>, %arg1: !torch.int, %arg2: !torch.bool, %arg3: !torch.bool, %arg4: !torch.optional<int>) -> !torch.int {\n"
" return %arg1 : !torch.int\n"
" }\n"
" func.func @\"__torch_mlir_dtype_fn.nvprims.convert_element_type\"(%arg0: !torch.tuple<int, int>, %arg1: !torch.int) -> !torch.int {\n"
" func.func @\"__torch_mlir_dtype_fn.prims.convert_element_type\"(%arg0: !torch.tuple<int, int>, %arg1: !torch.int) -> !torch.int {\n"
" return %arg1 : !torch.int\n"
" }\n"
" func.func @\"__torch_mlir_dtype_fn.aten.to.dtype_layout\"(%arg0: !torch.tuple<int, int>, %arg1: !torch.optional<int>, %arg2: !torch.optional<int>, %arg3: !torch.optional<Device>, %arg4: !torch.optional<bool>, %arg5: !torch.bool, %arg6: !torch.bool, %arg7: !torch.optional<int>) -> !torch.int {\n"

View File

@ -3196,7 +3196,7 @@ def atentodtype〡dtype(self_rank_dtype: Tuple[int, int], dtype: int, non_
_check_tensors_with_the_same_dtype(num_of_tensors=1, dtype=torch.float16) +
_check_tensors_with_the_same_dtype(num_of_tensors=1, dtype=torch.int32) +
_check_tensors_with_the_same_dtype(num_of_tensors=1, dtype=torch.complex64))
def nvprimsconvert_element_type〡dtype(a_rank_dtype: Tuple[int, int], dtype: int) -> int:
def primsconvert_element_type〡dtype(a_rank_dtype: Tuple[int, int], dtype: int) -> int:
return dtype
@check_dtype_function(_check_tensors_with_the_same_dtype(num_of_tensors=1) +

View File

@ -1 +1 @@
c01a41cdec4414d8853c8474ddcaf2bd6990e5c8
a099ff75d3e5ee1ae5ff1777ac52cb715183c6f8

View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch==2.1.0.dev20230808
torch==2.1.0.dev20230810

View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torchvision==0.16.0.dev20230808
torchvision==0.16.0.dev20230810