added fx tests

Author: msaroufim

CODEOWNERS

@@ -19,10 +19,12 @@ run_python_examples.sh @msaroufim
 
 # Need an owner 
 ./regression/
-./snli/ 
 ./mnist/
 ./imagenet/
 ./super_resolution/ 
 ./time_sequence_prediction/
 ./vae/
+
+# Legacy examples
 ./cpp/
+./legacy/snli/ 

snli/model.py legacy/snli/model.py

@@ -124,6 +124,20 @@ function snli() {
   python train.py --epochs 1 --dev_every 1 --no-bidirectional --dry-run || error "couldn't train snli"
 }
 
+function fx() {
+  start
+  python custom_tracer.py || error "fx custom tracer has failed"
+  python invert.py || error "fx invert has failed"
+  python module_tracer.py || error "fx module tracer has failed"
+  python nnc_compiler.py || error "fx nnc compile has failed"
+  python primitive_library.py || error "fx primitive library has failed"
+  python profiling_tracer.py || error "fx profiling tracer has failed"
+  python replace_op.py || error "fx replace op has failed"
+  python subgraph_rewriter_basic_use.py || error "fx subgraph has failed"
+  python vmap.py || error "vmap has failed"
+  python vmap_output_dynamically.py || error "vmap output dynamically has failed"
+}
+
 function super_resolution() {
   start
   python main.py --upscale_factor 3 --batchSize 4 --testBatchSize 100 --nEpochs 1 --lr 0.001  || error "super resolution failed"
@@ -185,6 +199,7 @@ function run_all() {
   time_sequence_prediction
   vae
   word_language_model
+  fx
 }
 
 # by default, run all examples