def dump_results(args):
print(" STATS ".center(72, '='))
fmt = "{2:>14}, {1:>8}, {0:<40}"
print(fmt.format("FUNCTION", "CALLS", "TIME"))
fmt = "{2:>14F}, {1:>8}, {0:<40}"
sorted_by_time = sorted(nvtx.getstats().iteritems(), key=lambda tup: tup[1][1])
for key, val in sorted_by_time:
print(fmt.format(key, *val))
args = pickle.load(f)
except EOFError:
break
with nvtx.profile_range('original'):
res_orig = cell_connectivity_original(*args)
with nvtx.profile_range('current in hexrd'):
res_current = cellConnectivity(*args)
with nvtx.profile_range('numba'):
res_numba = cell_connectivity_numba(*args)
with nvtx.profile_range('numba_bis'):
res_numba = cell_connectivity_numba(*args)
assert np.allclose(res_current, res_orig)
assert np.allclose(res_current, res_numba)
count += 1
if __name__ == '__main__':
in_file = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_ARG_FILE
run_test(in_file)
print(" STATS ".center(72, '='))
fmt = "{2:>14}, {1:>8}, {0:<40}"
print(fmt.format("FUNCTION", "CALLS", "TIME"))
fmt = "{2:>14F}, {1:>8}, {0:<40}"
sorted_by_time = sorted(nvtx.getstats().iteritems(),
key=lambda tup: tup[1][1])
for key, val in sorted_by_time:
print(fmt.format(key, *val))
with nvtx.profile_range('original'):
res_orig = cell_centroids_original(*args)
with nvtx.profile_range('current in hexrd'):
res_current = cellCentroids(*args)
with nvtx.profile_range('numba1'):
res_numba1 = cell_centroids_opt1(*args)
with nvtx.profile_range('numba1_bis'):
res_numba1 = cell_centroids_opt1(*args)
with nvtx.profile_range('numba'):
res_numba = cell_centroids_numba(*args)
with nvtx.profile_range('numba_bis'):
res_numba = cell_centroids_numba(*args)
assert np.allclose(res_current, res_orig)
assert np.allclose(res_current, res_numba1)
assert np.allclose(res_current, res_numba)
count += 1
if __name__=='__main__':
in_file = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_ARG_FILE
test_cellcentroids(in_file)
print(" STATS ".center(72, '='))
fmt = "{2:>14}, {1:>8}, {0:<40}"
print(fmt.format("FUNCTION", "CALLS", "TIME"))
fmt = "{2:>14F}, {1:>8}, {0:<40}"
sorted_by_time = sorted(nvtx.getstats().iteritems(), key=lambda tup: tup[1][1])
for key, val in sorted_by_time:
print(fmt.format(key, *val))
