示例#1
0
int main( int argc, char **argv )
{
	clock_t	t0 = StartTiming();

/* ---------- */
/* Parameters */
/* ---------- */

	MPIInit( argc, argv );

	if( !gArgs.SetCmdLine( argc, argv ) ||
		!gArgs.GetRanges() ) {

		MPIExit();
		exit( 42 );
	}

/* ------------ */
/* Initial data */
/* ------------ */

	if( !LayerCat( vL, gArgs.tempdir, gArgs.cachedir,
			gArgs.zolo, gArgs.zohi, false ) ) {

		MPIExit();
		exit( 42 );
	}

	InitTables( gArgs.zilo, gArgs.zihi );

	{
		CLoadPoints	*LP = new CLoadPoints;
		LP->Load( gArgs.tempdir, gArgs.cachedir );
		delete LP;
	}

/* ----- */
/* Solve */
/* ----- */

	printf( "\n---- Solve ----\n" );

	SetSolveParams( gArgs.regtype, gArgs.Wr, gArgs.Etol );

	XArray	Xevn, Xodd;

	if( !strcmp( gArgs.mode, "A2A" ) ) {

		Xevn.Load( gArgs.prior );

		if( gArgs.untwist )
			UntwistAffines( Xevn );

		Xodd.Resize( 6 );
		Solve( Xevn, Xodd, gArgs.iters );
	}
	else if( !strcmp( gArgs.mode, "A2H" ) ) {

		Xevn.Resize( 8 );

		{	// limit A lifetime
			XArray	*A = new XArray;
			A->Load( gArgs.prior );

			if( gArgs.untwist )
				UntwistAffines( *A );

			Solve( *A, Xevn, 1 );
			delete A;
		}

		Xodd.Resize( 8 );
		Solve( Xevn, Xodd, gArgs.iters );
	}
	else if( !strcmp( gArgs.mode, "H2H" ) ) {

		Xevn.Load( gArgs.prior );
		Xodd.Resize( 8 );
		Solve( Xevn, Xodd, gArgs.iters );
	}
	else if( !strcmp( gArgs.mode, "eval" ) ) {

		Xevn.Load( gArgs.prior );

		if( gArgs.untwist )
			UntwistAffines( Xevn );

		gArgs.iters = 0;
	}
	else {	// split

		Xevn.Load( gArgs.prior );
		gArgs.iters = 0;
	}

	const XArray& Xfinal = ((gArgs.iters & 1) ? Xodd : Xevn);

/* ----------- */
/* Postprocess */
/* ----------- */

	if( gArgs.mode[0] == 's' ) {

		Split	S( Xfinal, gArgs.splitmin );

		S.Run();
	}
	else {

		Evaluate( Xfinal );

		if( gArgs.mode[0] != 'e' )
			Xfinal.Save();
	}

/* ------- */
/* Cleanup */
/* ------- */

	if( !wkid ) {
		printf( "\n" );
		StopTiming( stdout, "Lsq", t0 );
	}

	MPIExit();
	VMStats( stdout );

	return 0;
}
示例#2
0
// Set wkid, and set nwks which determines if MPI is used,
// or not in the case that nwks == 1.
//
void MPIInit( int& argc, char**& argv )
{
// Look for the -nwks=j argument

    for( int i = 1; i < argc; ++i ) {
        if( GetArg( &nwks, "-nwks=%d", argv[i] ) )
            break;
    }

// Get wkid

    int	thrdtype = -1;

    if( nwks > 1 ) {

        // Start MPI

        MPI_Init_thread( &argc, &argv,
                         MPI_THREAD_FUNNELED, &thrdtype );

        MPI_Comm_rank( MPI_COMM_WORLD, &wkid );
    }

// Create log file 'lsqw_i'

    char slog[32];
    sprintf( slog, "lsqw_%d.txt", wkid );
    freopen( slog, "w", stdout );

    printf( "---- Read params ----\n" );

// MPI sanity checks

    if( nwks > 1 ) {

        // Verify worker count

        int	size;

        MPI_Comm_size( MPI_COMM_WORLD, &size );

        if( size != nwks ) {

            printf(
                "MPI: Bad worker count: %d, expected %d\n.",
                size, nwks );

            MPIExit();
            exit( 42 );
        }

        // Verify desired threading type.
        // Funneled means that the application can use
        // multiple threads but MPI communicates only
        // with the main thread.

        if( thrdtype != MPI_THREAD_FUNNELED ) {

            printf(
                "MPI: Not funneled thread type."
                " Try linking with -mt_mpi.\n" );

            MPIExit();
            exit( 42 );
        }
    }

    printf( "Worker %d / %d\n", wkid, nwks );
}