X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/kahina_paper2.git/blobdiff_plain/3991da6297471c9035632fb841e0490022b833a0..HEAD:/code.c?ds=sidebyside

diff --git a/code.c b/code.c
index cb93014..0059fc0 100644
--- a/code.c
+++ b/code.c
@@ -77,19 +77,21 @@ cuDoubleComplex NewH_EA(int i, cuDoubleComplex *Z) {
 
 //kernels to update a root i
 __global__
-cuDoubleComplex H_EA(int i, cuDoubleComplex *Z) {
-  cuDoubleComplex c;
-  //if the root needs to be updated
-  if(!finished[i]) {
-    //according to the module of the root
-    if (Cmodule(Z[i])<=maxRadius)
-      //selects the normal version
-      c = FirstH_EA(i,Z);  
-    else
-      //of the Log Exp version
-      c = NewH_EA(i,Z);
-    return c;
+void Dev_EA(int i, cuDoubleComplex *Z, int* finished,
+	    int size) {
+  int i= blockIdx.x*blockDim.x+ threadIdx.x;
+  if(i<size) {
+    //if the root needs to be updated
+    if(!finished[i]) {
+      //according to the module of the root
+      if (Cmodule(Z[i])<=maxRadius)
+	//selects the normal version
+	Z[i] = FirstH_EA(i,Z);  
+      else
+	//of the Log Exp version
+	Z[i] = NewH_EA(i,Z);
+      return c;
+    }
   }
-  else return Z[i];
 }