mirror of https://github.com/llvm/torch-mlir
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
parent
3854e9e5c3
commit
f0a8f273f7
|
@ -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"
|
||||
|
|
|
@ -3196,7 +3196,7 @@ def aten〇to〇dtype〡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 nvprims〇convert_element_type〡dtype(a_rank_dtype: Tuple[int, int], dtype: int) -> int:
|
||||
def prims〇convert_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) +
|
||||
|
|
|
@ -1 +1 @@
|
|||
c01a41cdec4414d8853c8474ddcaf2bd6990e5c8
|
||||
a099ff75d3e5ee1ae5ff1777ac52cb715183c6f8
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue