"openmmapi/src/PythonForce.cpp" did not exist on "54fa65ed0d443449b45a80bcddaba4894ffd9c41"
Commit 63f79469 authored by Chenggang Zhao's avatar Chenggang Zhao
Browse files

Renaming

parent d79b3cd1
...@@ -186,16 +186,16 @@ def bench_kineto(fn, kernel_names: Union[str, tuple], num_tests: int = 30, suppr ...@@ -186,16 +186,16 @@ def bench_kineto(fn, kernel_names: Union[str, tuple], num_tests: int = 30, suppr
if trace_path is not None: if trace_path is not None:
prof.export_chrome_trace(trace_path) prof.export_chrome_trace(trace_path)
# Return average kernel times # Return average kernel durations
units = {'ms': 1e3, 'us': 1e6} units = {'ms': 1e3, 'us': 1e6}
kernel_times = [] kernel_durations = []
for name in kernel_names: for name in kernel_names:
for line in prof_lines: for line in prof_lines:
if name in line: if name in line:
time_str = line.split()[-2] time_str = line.split()[-2]
for unit, scale in units.items(): for unit, scale in units.items():
if unit in time_str: if unit in time_str:
kernel_times.append(float(time_str.replace(unit, '')) / scale) kernel_durations.append(float(time_str.replace(unit, '')) / scale)
break break
break break
...@@ -211,11 +211,11 @@ def bench_kineto(fn, kernel_names: Union[str, tuple], num_tests: int = 30, suppr ...@@ -211,11 +211,11 @@ def bench_kineto(fn, kernel_names: Union[str, tuple], num_tests: int = 30, suppr
durations = [event['dur'] / 1e6 for event in events] durations = [event['dur'] / 1e6 for event in events]
assert len(durations) % num_kernels_per_period == 0 assert len(durations) % num_kernels_per_period == 0
num_kernel_patterns = len(durations) // num_kernels_per_period num_kernel_patterns = len(durations) // num_kernels_per_period
kernel_times[i] = [sum(durations[j::num_kernels_per_period]) / num_kernel_patterns kernel_durations[i] = [sum(durations[j::num_kernels_per_period]) / num_kernel_patterns
for j in range(num_kernels_per_period)] for j in range(num_kernels_per_period)]
# Return execution times # Return execution durations
return kernel_times if is_tuple else kernel_times[0] return kernel_durations if is_tuple else kernel_durations[0]
def hash_tensor(t: torch.Tensor): def hash_tensor(t: torch.Tensor):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment