Function main

deps/GraphBLAS/Demo/Program/complex_demo.c:68–134

//------------------------------------------------------------------------------

int main (int argc, char **argv)
{
    GrB_Index m = 3, k = 5, n = 4 ;
    GrB_Matrix A, B, C ;

    // initialize GraphBLAS and create the user-defined Complex type
    GrB_Info info ;
    GrB_init (GrB_NONBLOCKING) ;
    int nthreads ;
    GxB_Global_Option_get (GxB_GLOBAL_NTHREADS, &nthreads) ;
    fprintf (stderr, "complex_demo: nthreads: %d\n", nthreads) ;

    // print in 1-based notation
    GxB_Global_Option_set (GxB_PRINT_1BASED, true) ;

    bool predefined = (argc > 1) ;
    if (predefined)
    {
        fprintf (stderr, "Using pre-defined GxB_FC64 complex type\n") ;
    }
    else
    {
        fprintf (stderr, "Using user-defined Complex type\n") ;
    }

    info = Complex_init (predefined) ;
    if (info != GrB_SUCCESS)
    {
        fprintf (stderr, "Complex init failed\n") ;
        abort ( ) ;
    }

    // generate random matrices A and B
    simple_rand_seed (1) ;
    random_matrix (&A, false, false, m, k, 6, 0, true) ;
    random_matrix (&B, false, false, k, n, 8, 0, true) ;

    GxB_Matrix_fprint (A, "A", GxB_SHORT, stderr) ;
    GxB_Matrix_fprint (B, "B", GxB_SHORT, stderr) ;

    // C = A*B
    GrB_Matrix_new (&C, Complex, m, n) ;
    GrB_mxm (C, NULL, NULL, Complex_plus_times, A, B, NULL) ;

    GxB_Matrix_fprint (C, "C", GxB_SHORT, stderr) ;

    // print the results
    printf ("\n%% run this output of this program as a script:\n") ;
    print_complex_matrix (A, "A") ;
    print_complex_matrix (B, "B") ;
    print_complex_matrix (C, "C") ;

    printf ("E = A*B\n") ;
    printf ("err = norm (C-E,1)\n") ;
    printf ("assert (err < 1e-12)\n") ;

    // free all matrices
    GrB_Matrix_free (&A) ;