adding symmetric pattern option to the python interface

Author: liuyangzhuan

EXAMPLE/pddrive.py

@@ -75,7 +75,7 @@ def setup_pdbridge(sp,INT64):
 
 
 ####################### create the matrix
-INT64 = 0 # whether to use 64bit integer (requring superlu_dist to be compiled with 64-bit indexing)
+INT64 = 1 # whether to use 64bit integer (requring superlu_dist to be compiled with 64-bit indexing)
 rng = np.random.default_rng()
 n = 1000
 nrhs = 1
@@ -110,6 +110,7 @@ def setup_pdbridge(sp,INT64):
     argv.extend(['-q', '2'])  # column permutation 
     argv.extend(['-s', '0'])  # parallel symbolic factorization, needs -q to be 5
     argv.extend(['-i', '0'])  # whether to use iterative refinement 0, 1, 2
+    argv.extend(['-m', '0'])  # whether to use symmetric pattern 0 or 1
 argc = len(argv)
 if(rank==0):    
     print('SuperLU options: ',argv[1:])

SRC/double/pdbridge.c

@@ -32,7 +32,7 @@ void pdbridge_init(int_t m, int_t n, int_t nnz, int_t *rowind, int_t *colptr , d
     double   *berr;
     double   *b, *xtrue;
     int    m1, n1;
-    int      nprow, npcol, lookahead, colperm, rowperm, ir, symbfact, batch;
+    int      nprow, npcol, lookahead, colperm, rowperm, ir, symbfact, batch, sympattern;
     int      iam, info, ldb, ldx;
     char     **cpp, c, *postfix;;
     FILE *fp, *fopen();
@@ -56,6 +56,7 @@ void pdbridge_init(int_t m, int_t n, int_t nnz, int_t *rowind, int_t *colptr , d
     rowperm = -1;
     ir = -1;
     symbfact = -1;
+    sympattern=0;
     batch = 0;
 
     /* ------------------------------------------------------------
@@ -104,6 +105,7 @@ void pdbridge_init(int_t m, int_t n, int_t nnz, int_t *rowind, int_t *colptr , d
 	    switch (c) {
             case 'h':
                 printf("Options:\n");
+                printf("\t-m <int>: symmetric pattern  (default %4d)\n", sympattern);
                 printf("\t-r <int>: process rows       (default %4d)\n", nprow);
                 printf("\t-c <int>: process columns    (default %4d)\n", npcol);
                 printf("\t-p <int>: row permutation    (default %4d)\n", (slu_obj->options).RowPerm);
@@ -130,6 +132,8 @@ void pdbridge_init(int_t m, int_t n, int_t nnz, int_t *rowind, int_t *colptr , d
                     break;
             case 'b': batch = atoi(*cpp);
                     break;
+            case 'm': sympattern = atoi(*cpp);
+                    break;                    
 	    }
 	} else { /* Last arg is considered a filename */
 	    if ( !(fp = fopen(*cpp, "r")) ) {
@@ -145,6 +149,7 @@ void pdbridge_init(int_t m, int_t n, int_t nnz, int_t *rowind, int_t *colptr , d
     if (lookahead != -1) (slu_obj->options).num_lookaheads = lookahead;
     if (ir != -1) (slu_obj->options).IterRefine = ir;
     if (symbfact != -1) (slu_obj->options).ParSymbFact = symbfact;
+    if (sympattern==1) (slu_obj->options).SymPattern = YES;
 
     int superlu_acc_offload = sp_ienv_dist(10, &(slu_obj->options)); //get_acc_offload();
 

example_scripts/batch_script_mpi_runit_perlmutter_python_gcc_nvshmem.sh

@@ -93,6 +93,6 @@ export MPICH_MAX_THREAD_SAFETY=multiple
 
 
 cd ../EXAMPLE/
-srun -n $NCORE_VAL_TOT2D  -c $TH_PER_RANK --cpu_bind=cores python ./pddrive.py -c $NCOL -r $NROW 
+srun -n $NCORE_VAL_TOT2D  -c $TH_PER_RANK --cpu_bind=cores python ./pddrive.py -c $NCOL -r $NROW -s 1 -q 5 -m 1