From a0a249e460c98f22c4d23a6c9af6dc89ae20a63f Mon Sep 17 00:00:00 2001
From: Anthony Shoumikhin <shoumikhin@meta.com>
Date: Fri, 13 Sep 2024 10:37:48 -0700
Subject: [PATCH] Fix tests. (#5349)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/5349
overriding_review_checks_triggers_an_audit_and_retroactive_review
Oncall Short Name: executorch

Differential Revision: D62649178

fbshipit-source-id: 200a40b59f11ec5589a5abfef4d6b28b765cfe4a
---
 extension/module/test/module_test.cpp | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/extension/module/test/module_test.cpp b/extension/module/test/module_test.cpp
index 505040d7b3..f1871d631f 100644
--- a/extension/module/test/module_test.cpp
+++ b/extension/module/test/module_test.cpp
@@ -122,7 +122,7 @@ TEST_F(ModuleTest, TestNonExistentMethodMeta) {
 
 TEST_F(ModuleTest, TestExecute) {
   Module module(model_path_);
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.execute("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());
@@ -141,7 +141,7 @@ TEST_F(ModuleTest, TestExecutePreload) {
   const auto error = module.load();
   EXPECT_EQ(error, Error::Ok);
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.execute("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());
@@ -157,7 +157,7 @@ TEST_F(ModuleTest, TestExecutePreload_method) {
   const auto error = module.load_method("forward");
   EXPECT_EQ(error, Error::Ok);
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.execute("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());
@@ -176,7 +176,7 @@ TEST_F(ModuleTest, TestExecutePreloadProgramAndMethod) {
   const auto load_method_error = module.load_method("forward");
   EXPECT_EQ(load_method_error, Error::Ok);
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.execute("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());
@@ -205,7 +205,7 @@ TEST_F(ModuleTest, TestExecuteOnCurrupted) {
 TEST_F(ModuleTest, TestGet) {
   Module module(model_path_);
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.get("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());
@@ -280,7 +280,7 @@ TEST_F(ModuleTest, TestProgramSharingAndDataLoaderManagement) {
   EXPECT_EQ(load_error, Error::Ok);
   EXPECT_TRUE(module1->is_loaded());
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result1 = module1->execute("forward", {tensor, tensor});
   EXPECT_TRUE(result1.ok());
@@ -325,7 +325,7 @@ TEST_F(ModuleTest, TestProgramPersistenceAndReuseAfterModuleDestruction) {
 
   EXPECT_EQ(module.program(), shared_program);
 
-  auto tensor = make_tensor_ptr({1});
+  auto tensor = make_tensor_ptr({1}, {1});
 
   const auto result = module.execute("forward", {tensor, tensor});
   EXPECT_TRUE(result.ok());