Update et-pin.txt (pytorch#887)

* Update et-pin.txt Updating Pin for ET to 2024-07-01 nightly release * Update ManagedTensor to reflect the new API
yanbing-j · Jul 17, 2024 · 763c880 · 763c880
1 parent d1ae6b6
commit 763c880
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.pins/et-pin.txt b/.pins/et-pin.txt
@@ -1 +1 @@
-ff4e9edc55c86953ae33dbbaaacab8c9949dcb74
+172574a6be5910a4609e4ed1bef2b6b8475ddb3d
diff --git a/runner/run.cpp b/runner/run.cpp
@@ -203,9 +203,9 @@ float* forward(Transformer* transformer, int token, int pos) {
                              .to(torch::kCPU);
   auto logits = result[0].data_ptr();
 #else // __ET_MODEL__
-  ManagedTensor pos_managed(pos_buffer, sizeof(int64_t), {1}, ScalarType::Long);
+  ManagedTensor pos_managed(pos_buffer, {1}, ScalarType::Long);
   ManagedTensor tokens_managed(
-      token_buffer, sizeof(int64_t), {1, 1}, ScalarType::Long);
+      token_buffer, {1, 1}, ScalarType::Long);
   std::vector<EValue> inputs;
   auto tmp1 = EValue(tokens_managed.get_aliasing_tensor());
   auto tmp2 = EValue(pos_managed.get_aliasing_tensor());
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		ff4e9edc55c86953ae33dbbaaacab8c9949dcb74
		172574a6be5910a4609e4ed1bef2b6b8475ddb3d