Add continuous normalizing flow

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index 2cae095..291a7bb 100644
--- a/src/main.c
+++ b/src/main.c
@@ -6,6 +6,8 @@
 #include "train.h"
 #include "spiral.h"
 #include "tests.h"
+#include "test_cnf.h"
+#include "cnf_train.h"
 
 #include <stdio.h>
 #include <stdlib.h>
@@ -21,6 +23,14 @@ int main(void) {
     test_multi_obs_adjoint(&r);
     test_training(&r);
 
+    printf("\n--- CNF tests ---\n");
+    test_cnf_trace(&r);
+    test_cnf_invertibility(&r);
+    test_cnf_gradients(&r);
+    test_cnf_training(&r);
+
+    cnf_train_demo(&r);
+
     printf("\n--- Training demo (spiral) ---\n\n");
 
     r = rng_init((uint64_t)time(NULL));