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))