def test_write_non_distributed_npy(self):
cls = self.__class__
rank = cls.comm.getRank()
nproc = cls.comm.getSize()
if nproc > 1:
mvector_filename = 'out_mvector_write_test_' + str(nproc)
else:
mvector_filename = 'out_mvector_write_test'
file_dir = os.path.dirname(__file__)
filename = 'input.yaml'
problem = Utils.createAlbanyProblem(file_dir + '/' + filename,
cls.parallelEnv)
n_cols = 4
parameter_map = problem.getParameterMap(0)
mvector = Tpetra.MultiVector(parameter_map, n_cols, dtype="d")
mvector[0, :] = 1. * (rank + 1)
mvector[1, :] = -1. * (rank + 1)
mvector[2, :] = 3.26 * (rank + 1)
mvector[3, :] = -3.1 * (rank + 1)
Utils.writeMVector(file_dir + '/' + mvector_filename,
mvector,
distributedFile=False,
useBinary=True)
timers[1].start()
n_directions=4
parameter_map = problem.getParameterMap(0)
directions = Utils.loadMVector('random_directions', n_directions, parameter_map, distributedFile = False, useBinary = True)
timers[1].stop()
timers[2].start()
problem.setDirections(parameter_index, directions)
timers[2].stop()
timers[3].start()
problem.performSolve()
timers[3].stop()
timers[4].start()
hessian = problem.getReducedHessian(response_index, parameter_index)
timers[4].stop()
timers[5].start()
Utils.writeMVector("hessian_nprocs_"+str(nprocs), hessian, distributedFile = True, useBinary = False)
Utils.writeMVector("hessian_all_nprocs_"+str(nprocs), hessian, distributedFile = False, useBinary = False)
timers[5].stop()
print(hessian[0,0])
print(hessian[1,0])
print(hessian[2,0])
print(hessian[3,0])
timers[6].stop()
Utils.printTimers(timers, "timers_nprocs_"+str(nprocs)+".txt")
def main(parallelEnv):
# This example illustrates how PyAlbany can be used to compute
# reduced Hessian-vector products w.r.t to the basal friction.
comm = parallelEnv.comm
rank = comm.getRank()
nprocs = comm.getSize()
file_dir = os.path.dirname(__file__)
filename = 'input_fo_gis_analysis_beta_smbT.yaml'
parameter_index = 0
response_index = 0
timers = Utils.createTimers([
"PyAlbany: Create Albany Problem",
"PyAlbany: Read multivector directions", "PyAlbany: Set directions",
"PyAlbany: Perform Solve", "PyAlbany: Get Reduced Hessian",
"PyAlbany: Write Reduced Hessian", "PyAlbany: Total"
])
timers[6].start()
timers[0].start()
problem = Utils.createAlbanyProblem(filename, parallelEnv)
timers[0].stop()
timers[1].start()
n_directions = 4
parameter_map = problem.getParameterMap(0)
directions = Utils.loadMVector('random_directions',
n_directions,
parameter_map,
distributedFile=False,
useBinary=True)
timers[1].stop()
timers[2].start()
problem.setDirections(parameter_index, directions)
timers[2].stop()
timers[3].start()
problem.performSolve()
timers[3].stop()
timers[4].start()
hessian = problem.getReducedHessian(response_index, parameter_index)
timers[4].stop()
timers[5].start()
Utils.writeMVector("hessian_nprocs_" + str(nprocs),
hessian,
distributedFile=True,
useBinary=False)
Utils.writeMVector("hessian_all_nprocs_" + str(nprocs),
hessian,
distributedFile=False,
useBinary=False)
timers[5].stop()
print(hessian[0, 0])
print(hessian[1, 0])
print(hessian[2, 0])
print(hessian[3, 0])
timers[6].stop()
Utils.printTimers(timers, "timers_nprocs_" + str(nprocs) + ".txt")
