Skip to content

Commit 065a264

Browse files
authored
#437 Uplift mlir to d542487 and update toHost to expect vector return type (#438)
### Ticket #437 ### Problem description `tt::runtime::toHost` was updated in mlir commit 10c88b1 to return vector of tensors, which breaks torch build ### What's changed Updated to use first tensor returned in list of tensors by toHost ### Checklist - [x] New/Existing tests provide coverage for changes
1 parent f0fc4ae commit 065a264

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

third_party/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
# SPDX-License-Identifier: Apache-2.0
44
#
55

6-
set(TT_MLIR_VERSION "2d38c0af771c191ff2fe34469926e5c455016ffd")
6+
set(TT_MLIR_VERSION "d5424879e434711388f17cbc881e437b0b84ca1f")
77
set(TORCH_MLIR_VERSION "a265d283357f572c5437f9217ea85fa9770d374c")
88

99
if (BUILD_TOOLCHAIN)

tt_torch/csrc/bindings.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ std::vector<int64_t> as_vec_int64(std::vector<T> const &vec) {
109109
static torch::Tensor create_torch_tensor(const tt::runtime::Tensor &tensor,
110110
const tt::runtime::TensorDesc &desc) {
111111
tt::runtime::Tensor untilized_tensor =
112-
tt::runtime::toHost(tensor, /*untilize=*/true);
112+
tt::runtime::toHost(tensor, /*untilize=*/true)[0];
113113
const std::vector<std::int64_t> shape = as_vec_int64(desc.shape);
114114
const std::vector<std::int64_t> stride = as_vec_int64((desc.stride));
115115

0 commit comments

Comments
 (0)