Add options for user to input supernode partition.

Author: xiaoyeli

SRC/include/superlu_defs.h

@@ -746,6 +746,7 @@ typedef struct {
     double	  ILU_FillTol;    /* threshold for zero pivot perturbation */
     milu_t	  ILU_MILU;
     double	  ILU_MILU_Dim;   /* Dimension of PDE (if available) */
+    yes_no_t      UserDefineSupernode; /* Allow users to define supernodes */
     yes_no_t      ParSymbFact;
     yes_no_t      ReplaceTinyPivot; /* used in SuperLU_DIST */
     yes_no_t      SolveInitialized;

SRC/prec-independent/ilu_level_symbfact.c

@@ -64,15 +64,6 @@ int_t ilu_level_symbfact
     double t;
     int min_mn = SUPERLU_MIN(m, n);
 
-    /* Set up supernode partition */
-    Glu_persist->supno = (int_t *) SUPERLU_MALLOC(n * sizeof(int_t));
-    Glu_persist->xsup = (int_t *) SUPERLU_MALLOC((n+1) * sizeof(int_t));
-    for (i = 0; i < n; i++) {
-	Glu_persist->supno[i] = i;
-	Glu_persist->xsup[i] = i;
-    }
-    Glu_persist->xsup[n] = n;
-
 
     NCPformat *GACstore = A->Store;
     int_t *colbeg, *colend, *rowind, irow;
@@ -84,6 +75,17 @@ int_t ilu_level_symbfact
     Glu_freeable->xusub = (int_t *) intCalloc_dist(n+1);
     int_t nnzL = 0, nnzU = 0;
 
+    /* Set up supernode partition */
+    Glu_persist->supno = (int_t *) SUPERLU_MALLOC(n * sizeof(int_t));
+    Glu_persist->xsup = (int_t *) SUPERLU_MALLOC((n+1) * sizeof(int_t));
+    if ( options->UserDefineSupernode == NO ) {
+	for (i = 0; i < n; i++) { /* set up trivial supernode for now. */
+	    Glu_persist->supno[i] = i;
+	    Glu_persist->xsup[i] = i;
+	}
+	Glu_persist->xsup[n] = n;
+    }
+
     /* Count nonzeros per column for L & U */
     for (int j = 0; j < n; ++j) {
 #if 0	
@@ -135,6 +137,7 @@ int_t ilu_level_symbfact
 	Glu_freeable->xusub[i] += Glu_freeable->xusub[i-1];
     }
 
+
     return ( -(Glu_freeable->xlsub[n] + Glu_freeable->xusub[n]) );
 }
 

SRC/prec-independent/util.c

@@ -210,6 +210,7 @@ void set_default_options_dist(superlu_dist_options_t *options)
     options->Fact = DOFACT;
     options->Equil = YES;
     options->ILU_level = SLU_EMPTY;
+    options->UserDefineSupernode = NO; /* detect supernodes internally */
     options->ParSymbFact = NO;
 #ifdef HAVE_PARMETIS
     options->ColPerm = METIS_AT_PLUS_A;
@@ -256,6 +257,7 @@ void print_options_dist(superlu_dist_options_t *options)
     printf("**    Fact                      : %4d\n", options->Fact);
     printf("**    Equil                     : %4d\n", options->Equil);
     printf("**    DiagInv                   : %4d\n", options->DiagInv);
+    printf("**    UserDefineSupernode       : %4d\n", options->UserDefineSupernode);
     printf("**    ParSymbFact               : %4d\n", options->ParSymbFact);
     printf("**    ColPerm                   : %4d\n", options->ColPerm);
     printf("**    RowPerm                   : %4d\n", options->RowPerm);