Write out K

1 files changed, 10 insertions, 1 deletions

diff --git a/src/gemma.cpp b/src/gemma.cpp
index 6b42763..1cced77 100644
--- a/src/gemma.cpp
+++ b/src/gemma.cpp
@@ -1915,7 +1915,16 @@ void GEMMA::BatchRun(PARAM &cPar) {
     cout << "Calculating Relatedness Matrix ... " << endl;
 
     if (cPar.is_mdb) {
-      cPar.MdbCalcKin();
+      time_start = clock();
+      auto K = cPar.MdbCalcKin();
+      cPar.time_G = (clock() - time_start) / (double(CLOCKS_PER_SEC) * 60.0);
+      if (cPar.a_mode == M_KIN) {
+        cPar.WriteMatrix(K, "cXX");
+      } else { // M_KIN2
+        cPar.WriteMatrix(K, "sXX");
+      }
+
+      gsl_matrix_safe_free(K);
       return;
     }