fp32

Author: kirklandsign

extension/android/executorch_android/android_test_setup.sh

@@ -25,7 +25,7 @@ prepare_tinyllama() {
   # Create params.json file
   touch params.json
   echo '{"dim": 288, "multiple_of": 32, "n_heads": 6, "n_layers": 6, "norm_eps": 1e-05, "vocab_size": 32000}' > params.json
-  python -m examples.models.llama.export_llama -c stories15M.pt -p params.json -d fp16 -n stories15m_h.pte -kv
+  python -m examples.models.llama.export_llama -c stories15M.pt -p params.json -d fp32 -n stories15m_h.pte -kv
   python -m pytorch_tokenizers.tools.llama2c.convert -t tokenizer.model -o tokenizer.bin
 
   cp stories15m_h.pte "${BASEDIR}/src/androidTest/resources/stories.pte"

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

@@ -99,6 +99,7 @@ public void testStories() throws IOException, URISyntaxException{
 
         Module module = Module.load(getTestFilePath(filePath));
         module.loadMethod("forward");
+        module.forward();
     }
 
     public void testMv2Fp32() throws IOException, URISyntaxException {