mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
12cb26509a
This is a big bang PR, merge conflicts are probably expected and will be addressed at merge. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81643 Approved by: https://github.com/ezyang
22 lines
518 B
Python
22 lines
518 B
Python
import torch._C._lazy
|
|
|
|
|
|
def reset():
|
|
"""Resets all metric counters."""
|
|
torch._C._lazy._reset_metrics()
|
|
|
|
|
|
def counter_names():
|
|
"""Retrieves all the currently active counter names."""
|
|
return torch._C._lazy._counter_names()
|
|
|
|
|
|
def counter_value(name: str):
|
|
"""Return the value of the counter with the speficied name"""
|
|
return torch._C._lazy._counter_value(name)
|
|
|
|
|
|
def metrics_report():
|
|
"""Return the combined (lazy core and backend) metric report"""
|
|
return torch._C._lazy._metrics_report()
|