Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 0cc6d7c

Browse files
committed
tensor_ptr arg order change
1 parent 276073e commit 0cc6d7c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

runner/run.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,8 +213,8 @@ float* forward(Transformer* transformer, int token, int pos) {
213213
.to(torch::kCPU);
214214
auto logits = result[0].data_ptr();
215215
#else // __ET_MODEL__
216-
TensorPtr pos_managed = make_tensor_ptr(ScalarType::Long, {1}, pos_buffer);
217-
TensorPtr tokens_managed = make_tensor_ptr(ScalarType::Long, {1, 1}, token_buffer);
216+
TensorPtr pos_managed = make_tensor_ptr({1}, pos_buffer, ScalarType::Long);
217+
TensorPtr tokens_managed = make_tensor_ptr({1, 1}, token_buffer, ScalarType::Long);
218218
std::vector<EValue> inputs;
219219
auto tmp1 = EValue(tokens_managed);
220220
auto tmp2 = EValue(pos_managed);

0 commit comments

Comments
 (0)