Skip to content

Commit 32e8b53

Browse files
committed
fix: Use scripting instead of tracing for module fallback tests
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 282e98a commit 32e8b53

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: tests/cpp/test_module_fallback.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
TEST(CppAPITest, ResNetModuleFallbacksCorrectly) {
88
torch::jit::script::Module mod;
99
try {
10-
mod = torch::jit::load("tests/modules/resnet18_traced.jit.pt");
10+
mod = torch::jit::load("tests/modules/resnet18_scripted.jit.pt");
1111
} catch (const c10::Error& e) {
1212
std::cerr << "error loading the model\n";
1313
ASSERT_TRUE(false);
@@ -35,7 +35,7 @@ TEST(CppAPITest, ResNetModuleFallbacksCorrectly) {
3535
TEST(CppAPITest, MobileNetModuleFallbacksCorrectlyWithOneEngine) {
3636
torch::jit::script::Module mod;
3737
try {
38-
mod = torch::jit::load("tests/modules/mobilenet_v2_traced.jit.pt");
38+
mod = torch::jit::load("tests/modules/mobilenet_v2_scripted.jit.pt");
3939
} catch (const c10::Error& e) {
4040
std::cerr << "error loading the model\n";
4141
ASSERT_TRUE(false);

0 commit comments

Comments
 (0)