Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
DeepEP
Commits
63f79469
"openmmapi/src/PythonForce.cpp" did not exist on "54fa65ed0d443449b45a80bcddaba4894ffd9c41"
Commit
63f79469
authored
Jul 02, 2025
by
Chenggang Zhao
Browse files
Renaming
parent
d79b3cd1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
tests/utils.py
tests/utils.py
+6
-6
No files found.
tests/utils.py
View file @
63f79469
...
@@ -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
time
s
# Return average kernel
duration
s
units
=
{
'ms'
:
1e3
,
'us'
:
1e6
}
units
=
{
'ms'
:
1e3
,
'us'
:
1e6
}
kernel_
time
s
=
[]
kernel_
duration
s
=
[]
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_
time
s
.
append
(
float
(
time_str
.
replace
(
unit
,
''
))
/
scale
)
kernel_
duration
s
.
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_
time
s
[
i
]
=
[
sum
(
durations
[
j
::
num_kernels_per_period
])
/
num_kernel_patterns
kernel_
duration
s
[
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
time
s
# Return execution
duration
s
return
kernel_
time
s
if
is_tuple
else
kernel_
time
s
[
0
]
return
kernel_
duration
s
if
is_tuple
else
kernel_
duration
s
[
0
]
def
hash_tensor
(
t
:
torch
.
Tensor
):
def
hash_tensor
(
t
:
torch
.
Tensor
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment