示例#1
0
def wait_for_keypress(discr):
    """MPI-aware keypress wait"""
    try:
        comm = discr.parallel_discr.context.communicator
    except AttributeError:
        raw_input("[Enter]")
    else:
        if comm.rank == 0:
            # OpenMPI connects mpirun's stdin to rank 0's stdin.
            print "[Enter]"
            raw_input()

        from boostmpi import broadcast
        broadcast(comm, value=0, root=0)
示例#2
0
def wait_for_keypress(discr):
    """MPI-aware keypress wait"""
    try:
        comm = discr.parallel_discr.context.communicator
    except AttributeError:
        raw_input("[Enter]")
    else:
        if comm.rank == 0:
            # OpenMPI connects mpirun's stdin to rank 0's stdin.
            print "[Enter]"
            raw_input()

        from boostmpi import broadcast
        broadcast(comm, value=0, root=0)
示例#3
0
#!/usr/bin/env python

import boostmpi as mpi
import sys

x1, x2, x3 = None, None, None

if mpi.rank == 0:
    x1, x2, x3 = 1.2, 1.3, 1.4
    #ans = raw_input('Question? ')
    #print ans

#mpi.world.barrier()
x1, x2, x3 = mpi.broadcast(mpi.world, root=0, value=(x1, x2, x3))

print mpi.rank, x1, x2, x3

mpi.world.barrier()
if mpi.rank == 0:
    print 'Question? (Y/n) '
    ans = sys.stdin.readline()[:-1]
    print ans
            h5group_ngpu = fd.create_group(grp_name)
        except ValueError:
            print "'/%s' group is already present." % grp_name
            print 'Do you want to remove the existing group? (y/N)'
            ans = sys.stdin.readline()
            if ans == 'Y\n' or ans == 'y\n':
                del fd[grp_name]
                h5group_ngpu = fd.create_group(grp_name)
            elif ans == 'N\n' or ans == 'n\n' or ans == '\n':
                h5group_ngpu = fd[grp_name]
            else:
                sys.exit()


    m_list, num_float, num_gpus, num_avg, transfer_type_list = \
      mpi.broadcast( mpi.world, root=0, value=(m_list, num_float, num_gpus, num_avg, transfer_type_list) )
    multi_obj = BandwidthTestMulti(m_list, num_float, num_gpus, num_avg)

    if 'serial' in transfer_type_list:
        if mpi.rank == 0:
            print 'GPU-PCIE-bandwidth test: serial'
            multi_obj.set_h5group_ngpu(h5group_ngpu)
            t1 = datetime.now()
            multi_obj.serial()
            dt = datetime.now() - t1
            total_dt[0] = dt.seconds + dt.microseconds * 1e-6
            print 'serial test end.', dt

    if 'mpi' in transfer_type_list:
        if mpi.rank == 0:
            print 'GPU-PCIE-bandwidth test: mpi'
#!/usr/bin/env python

import boostmpi as mpi
import sys

x1, x2, x3 = None, None, None

if mpi.rank == 0:
	x1, x2, x3 = 1.2, 1.3, 1.4
	#ans = raw_input('Question? ')
	#print ans

#mpi.world.barrier()
x1, x2, x3 = mpi.broadcast( mpi.world, root=0, value=(x1, x2, x3) )

print mpi.rank, x1, x2, x3

mpi.world.barrier()
if mpi.rank == 0:
	print 'Question? (Y/n) '
	ans = sys.stdin.readline()[:-1]
	print ans
            grp_name = "%dgpu" % num_gpus
            h5group_ngpu = fd.create_group(grp_name)
        except ValueError:
            print "'/%s' group is already present." % grp_name
            print "Do you want to remove the existing group? (y/N)"
            ans = sys.stdin.readline()
            if ans == "Y\n" or ans == "y\n":
                del fd[grp_name]
                h5group_ngpu = fd.create_group(grp_name)
            elif ans == "N\n" or ans == "n\n" or ans == "\n":
                h5group_ngpu = fd[grp_name]
            else:
                sys.exit()

    m_list, num_float, num_gpus, num_avg, transfer_type_list = mpi.broadcast(
        mpi.world, root=0, value=(m_list, num_float, num_gpus, num_avg, transfer_type_list)
    )
    multi_obj = BandwidthTestMulti(m_list, num_float, num_gpus, num_avg)

    if "serial" in transfer_type_list:
        if mpi.rank == 0:
            print "GPU-PCIE-bandwidth test: serial"
            multi_obj.set_h5group_ngpu(h5group_ngpu)
            t1 = datetime.now()
            multi_obj.serial()
            dt = datetime.now() - t1
            total_dt[0] = dt.seconds + dt.microseconds * 1e-6
            print "serial test end.", dt

    if "mpi" in transfer_type_list:
        if mpi.rank == 0: