From 3fe15e5e5c591463309b38cc2ba3b77787814714 Mon Sep 17 00:00:00 2001
From: Alejandro Gallo <aamsgallo@gmail.com>
Date: Mon, 23 Jan 2023 16:57:07 +0100
Subject: [PATCH] Fix bs and ths error in equations

---
 src/atrip/Equations.cxx | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/atrip/Equations.cxx b/src/atrip/Equations.cxx
index 9f42613..b4c19df 100644
--- a/src/atrip/Equations.cxx
+++ b/src/atrip/Equations.cxx
@@ -533,11 +533,15 @@ void getEnergySame
 //                        cuMemAlloc((CUdeviceptr*)&_vhhh,
 //                                   NoNoNo * sizeof(DataFieldType<F>)));
 //                )
-//    const size_t
-//      bs = Atrip::kernelDimensions.ooo.blocks,
-//      ths = Atrip::kernelDimensions.ooo.threads;
-    //cuda::zeroing<<<bs, ths>>>((DataFieldType<F>*)_t_buffer, NoNoNo);
-    //cuda::zeroing<<<bs, ths>>>((DataFieldType<F>*)_vhhh, NoNoNo);
+#if !defined(ATRIP_ONLY_DGEMM)
+    // we still have to zero this
+    const size_t
+      bs = Atrip::kernelDimensions.ooo.blocks,
+      ths = Atrip::kernelDimensions.ooo.threads;
+    cuda::zeroing<<<bs, ths>>>((DataFieldType<F>*)_t_buffer, NoNoNo);
+    cuda::zeroing<<<bs, ths>>>((DataFieldType<F>*)_vhhh, NoNoNo);
+#endif
+
 #else
     DataFieldType<F>* _t_buffer = (DataFieldType<F>*)malloc(NoNoNo * sizeof(F));
     DataFieldType<F>* _vhhh = (DataFieldType<F>*)malloc(NoNoNo * sizeof(F));