Skip to content

Commit

Permalink
Fix tests. (#5349)
Browse files Browse the repository at this point in the history
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
  • Loading branch information
shoumikhin authored and facebook-github-bot committed Sep 13, 2024
1 parent 37f77ed commit a0a249e
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions extension/module/test/module_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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());
Expand All @@ -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());
Expand All @@ -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());
Expand All @@ -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());
Expand Down Expand Up @@ -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());
Expand Down Expand Up @@ -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());
Expand Down Expand Up @@ -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());
Expand Down

0 comments on commit a0a249e

Please sign in to comment.