and just load

Author: kirklandsign

extension/android/executorch_android/src/androidTest/java/org/pytorch/executorch/ModuleE2ETest.java

@@ -59,4 +59,18 @@ public void testMv2Fp32() throws IOException, URISyntaxException{
         assertTrue(results[0].isTensor());
     }
 
+    @Test
+    public void testStories() throws IOException, URISyntaxException{
+        String filePath = "/stories.pte";
+        File pteFile = new File(getTestFilePath(filePath));
+        InputStream inputStream = getClass().getResourceAsStream(filePath);
+        FileUtils.copyInputStreamToFile(inputStream, pteFile);
+        inputStream.close();
+
+        Module module = Module.load(getTestFilePath(filePath));
+
+        EValue[] results = module.forward();
+    }
+
+
 }

scripts/run_android_emulator.sh

@@ -25,7 +25,7 @@ done &
 
 adb logcat -c
 adb shell am instrument -w -r -e \
-  class org.pytorch.executorch.LlmModuleInstrumentationTest,org.pytorch.executorch.ModuleE2ETest \
+  class org.pytorch.executorch.ModuleInstrumentationTest,org.pytorch.executorch.ModuleE2ETest \
   org.pytorch.executorch.test/androidx.test.runner.AndroidJUnitRunner >result.txt 2>&1
 adb logcat -d > logcat.txt
 adb shell dumpsys meminfo