Skip to content

Commit 0f3bbec

Browse files
Jack-Khuumalfet
authored andcommitted
Update et-pin.txt (#887)
* Update et-pin.txt Updating Pin for ET to 2024-07-01 nightly release * Update ManagedTensor to reflect the new API
1 parent 58b1eaf commit 0f3bbec

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.pins/et-pin.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
ff4e9edc55c86953ae33dbbaaacab8c9949dcb74
1+
172574a6be5910a4609e4ed1bef2b6b8475ddb3d

runner/run.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,9 +203,9 @@ float* forward(Transformer* transformer, int token, int pos) {
203203
.to(torch::kCPU);
204204
auto logits = result[0].data_ptr();
205205
#else // __ET_MODEL__
206-
ManagedTensor pos_managed(pos_buffer, sizeof(int64_t), {1}, ScalarType::Long);
206+
ManagedTensor pos_managed(pos_buffer, {1}, ScalarType::Long);
207207
ManagedTensor tokens_managed(
208-
token_buffer, sizeof(int64_t), {1, 1}, ScalarType::Long);
208+
token_buffer, {1, 1}, ScalarType::Long);
209209
std::vector<EValue> inputs;
210210
auto tmp1 = EValue(tokens_managed.get_aliasing_tensor());
211211
auto tmp2 = EValue(pos_managed.get_aliasing_tensor());

0 commit comments

Comments
 (0)