From 32e8b5390de7447060f8e9ba81383f8c1138ef85 Mon Sep 17 00:00:00 2001
From: Dheeraj Peri <peri.dheeraj@gmail.com>
Date: Sat, 9 Oct 2021 09:06:57 -0700
Subject: [PATCH] fix: Use scripting instead of tracing for module fallback
 tests

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
---
 tests/cpp/test_module_fallback.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/cpp/test_module_fallback.cpp b/tests/cpp/test_module_fallback.cpp
index d74d75d5e6..f6f101b7db 100644
--- a/tests/cpp/test_module_fallback.cpp
+++ b/tests/cpp/test_module_fallback.cpp
@@ -7,7 +7,7 @@
 TEST(CppAPITest, ResNetModuleFallbacksCorrectly) {
   torch::jit::script::Module mod;
   try {
-    mod = torch::jit::load("tests/modules/resnet18_traced.jit.pt");
+    mod = torch::jit::load("tests/modules/resnet18_scripted.jit.pt");
   } catch (const c10::Error& e) {
     std::cerr << "error loading the model\n";
     ASSERT_TRUE(false);
@@ -35,7 +35,7 @@ TEST(CppAPITest, ResNetModuleFallbacksCorrectly) {
 TEST(CppAPITest, MobileNetModuleFallbacksCorrectlyWithOneEngine) {
   torch::jit::script::Module mod;
   try {
-    mod = torch::jit::load("tests/modules/mobilenet_v2_traced.jit.pt");
+    mod = torch::jit::load("tests/modules/mobilenet_v2_scripted.jit.pt");
   } catch (const c10::Error& e) {
     std::cerr << "error loading the model\n";
     ASSERT_TRUE(false);