run aoti

Author: Michael Gschwind

.github/workflows/compile.yml

@@ -37,9 +37,10 @@ jobs:
       - name: Run inference
         run: |          
           export MODEL_PATH=checkpoints/stories15M/stories15M.pt
-          # python aoti_export.py --checkpoint_path ${MODEL_PATH} --output_path ./${MODEL_REPO}.so
+	  export MODEL_REPO=stories15M
           python generate.py --checkpoint_path ${MODEL_PATH} --temperature 0 --device cpu | tee output_eager
           python generate.py --compile --checkpoint_path ${MODEL_PATH} --temperature 0 --device cpu | tee output_compiled
+          python aoti_export.py --checkpoint_path ${MODEL_PATH} --output_path ./${MODEL_REPO}.so
           python generate.py --checkpoint_path ${MODEL_PATH} --temperature 0 --dso ./${MODEL_REPO}.so | tee output_aoti
           # echo "******************************************"
           # echo "********* EAGER vs TORCH.COMPILE *********"