From fc242f52e835be85cc6030b6cae5619d18df7670 Mon Sep 17 00:00:00 2001
From: Jaron Kent-Dobias <jaron@kent-dobias.com>
Date: Mon, 30 Jul 2018 06:32:51 -0400
Subject: various changes

---
 lib/correlation.h | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'lib/correlation.h')

diff --git a/lib/correlation.h b/lib/correlation.h
index d08f713..29357a5 100644
--- a/lib/correlation.h
+++ b/lib/correlation.h
@@ -10,7 +10,11 @@ template <class R_t, class X_t>
 double correlation_length(const state_t <R_t, X_t>& s) {
   double total = 0;
 
+#ifdef DIMENSION
+  for (D_t j = 0; j < DIMENSION; j++) {
+#else
   for (D_t j = 0; j < s.D; j++) {
+#endif
     total += norm_squared(s.ReF[j]) + norm_squared(s.ImF[j]);
   }
 
-- 
cgit v1.2.3-70-g09d2