Skip to content

Commit a0a249e

Browse files
shoumikhinfacebook-github-bot
authored andcommitted
Fix tests. (#5349)
Summary: Pull Request resolved: #5349 overriding_review_checks_triggers_an_audit_and_retroactive_review Oncall Short Name: executorch Differential Revision: D62649178 fbshipit-source-id: 200a40b59f11ec5589a5abfef4d6b28b765cfe4a
1 parent 37f77ed commit a0a249e

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

extension/module/test/module_test.cpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ TEST_F(ModuleTest, TestNonExistentMethodMeta) {
122122

123123
TEST_F(ModuleTest, TestExecute) {
124124
Module module(model_path_);
125-
auto tensor = make_tensor_ptr({1});
125+
auto tensor = make_tensor_ptr({1}, {1});
126126

127127
const auto result = module.execute("forward", {tensor, tensor});
128128
EXPECT_TRUE(result.ok());
@@ -141,7 +141,7 @@ TEST_F(ModuleTest, TestExecutePreload) {
141141
const auto error = module.load();
142142
EXPECT_EQ(error, Error::Ok);
143143

144-
auto tensor = make_tensor_ptr({1});
144+
auto tensor = make_tensor_ptr({1}, {1});
145145

146146
const auto result = module.execute("forward", {tensor, tensor});
147147
EXPECT_TRUE(result.ok());
@@ -157,7 +157,7 @@ TEST_F(ModuleTest, TestExecutePreload_method) {
157157
const auto error = module.load_method("forward");
158158
EXPECT_EQ(error, Error::Ok);
159159

160-
auto tensor = make_tensor_ptr({1});
160+
auto tensor = make_tensor_ptr({1}, {1});
161161

162162
const auto result = module.execute("forward", {tensor, tensor});
163163
EXPECT_TRUE(result.ok());
@@ -176,7 +176,7 @@ TEST_F(ModuleTest, TestExecutePreloadProgramAndMethod) {
176176
const auto load_method_error = module.load_method("forward");
177177
EXPECT_EQ(load_method_error, Error::Ok);
178178

179-
auto tensor = make_tensor_ptr({1});
179+
auto tensor = make_tensor_ptr({1}, {1});
180180

181181
const auto result = module.execute("forward", {tensor, tensor});
182182
EXPECT_TRUE(result.ok());
@@ -205,7 +205,7 @@ TEST_F(ModuleTest, TestExecuteOnCurrupted) {
205205
TEST_F(ModuleTest, TestGet) {
206206
Module module(model_path_);
207207

208-
auto tensor = make_tensor_ptr({1});
208+
auto tensor = make_tensor_ptr({1}, {1});
209209

210210
const auto result = module.get("forward", {tensor, tensor});
211211
EXPECT_TRUE(result.ok());
@@ -280,7 +280,7 @@ TEST_F(ModuleTest, TestProgramSharingAndDataLoaderManagement) {
280280
EXPECT_EQ(load_error, Error::Ok);
281281
EXPECT_TRUE(module1->is_loaded());
282282

283-
auto tensor = make_tensor_ptr({1});
283+
auto tensor = make_tensor_ptr({1}, {1});
284284

285285
const auto result1 = module1->execute("forward", {tensor, tensor});
286286
EXPECT_TRUE(result1.ok());
@@ -325,7 +325,7 @@ TEST_F(ModuleTest, TestProgramPersistenceAndReuseAfterModuleDestruction) {
325325

326326
EXPECT_EQ(module.program(), shared_program);
327327

328-
auto tensor = make_tensor_ptr({1});
328+
auto tensor = make_tensor_ptr({1}, {1});
329329

330330
const auto result = module.execute("forward", {tensor, tensor});
331331
EXPECT_TRUE(result.ok());

0 commit comments

Comments
 (0)