Put grammar stuff.

diff --git a/src/main.ml b/src/main.ml
index fa16737..42c8822 100644 (file)
--- a/src/main.ml
+++ b/src/main.ml
@@ -126,10 +126,16 @@ let _ =
          Gc.compact();
          Gc.set (tuned_gc);
          let runtime =
-           let module R = ResJIT.Count in
+           if !Options.count_only then
+           let module R = ResJIT.Make(NodeSet.Partial(NodeSet.Count)) in
            let module M = Runtime.Make(R) in
        (* mk_runtime run auto doc arg count print outfile  *)
            mk_runtime M.grammar_run auto (Obj.magic g) () R.NS.length (Obj.magic R.NS.serialize) None
+           else
+             let module R = ResJIT.Mat in
+             let module M = Runtime.Make(R) in
+           (* mk_runtime run auto doc arg count print outfile  *)
+             mk_runtime M.grammar_run auto (Obj.magic g) () R.NS.length (Obj.magic R.NS.serialize) None
          in
          runtime ();
          exit 0