just load

Author: kirklandsign

.github/workflows/_android.yml

@@ -65,6 +65,7 @@ jobs:
   run-emulator:
     # NB: Use metal install for KVM support to run the emulator faster
     runs-on: linux.24xl.spr-metal
+    needs: build-llm-demo
     env:
       ANDROID_NDK_VERSION: r27b
       API_LEVEL: 34

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

@@ -74,42 +74,10 @@ public void setUp() throws IOException {
     public GrantPermissionRule mRuntimePermissionRule = GrantPermissionRule.grant(Manifest.permission.READ_EXTERNAL_STORAGE);
 
     @Test
-    public void testGenerate() throws IOException, URISyntaxException{
+    public void testLoad() throws IOException, URISyntaxException{
         int loadResult = mModule.load();
         // Check that the model can be load successfully
         assertEquals(OK, loadResult);
-
-        mModule.generate(TEST_PROMPT, SEQ_LEN, LlmModuleInstrumentationTest.this);
-        assertEquals(results.size(), SEQ_LEN);
-        assertTrue(tokensPerSecond.get(tokensPerSecond.size() - 1) > 0);
     }
 
-    @Test
-    public void testGenerateAndStop() throws IOException, URISyntaxException{
-        mModule.generate(TEST_PROMPT, SEQ_LEN, new LlmCallback() {
-            @Override
-            public void onResult(String result) {
-                LlmModuleInstrumentationTest.this.onResult(result);
-                mModule.stop();
-            }
-
-            @Override
-            public void onStats(float tps) {
-                LlmModuleInstrumentationTest.this.onStats(tps);
-            }
-        });
-
-        int stoppedResultSize = results.size();
-        assertTrue(stoppedResultSize < SEQ_LEN);
-    }
-
-    @Override
-    public void onResult(String result) {
-        results.add(result);
-    }
-
-    @Override
-    public void onStats(float tps) {
-        tokensPerSecond.add(tps);
-    }
 }