Skip to content

Commit

Permalink
Log the optimus graph transformation to the scuba
Browse files Browse the repository at this point in the history
Summary: Current everstore upload logging may cuase excessive compilation time when the model has lots of graph breaks (post: https://fb.workplace.com/groups/257735836456307/permalink/633533465543207/), we here log the transformation only when the graph changed

Differential Revision: D53692344
  • Loading branch information
mengluy0125 authored and facebook-github-bot committed Feb 14, 2024
1 parent dbc1097 commit 4d23fcd
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,7 @@


counters: DefaultDict[str, Counter[str]] = collections.defaultdict(collections.Counter)
optimus_scuba_log: Dict[str, Any] = {}
troubleshooting_url = "https://pytorch.org/docs/master/compile/troubleshooting.html"
nnmodule_doc_url = "https://pytorch.org/docs/master/compile/nn-module.html"
nnmodule_doc_url_msg = f"See {nnmodule_doc_url} for more information and limitations."
Expand Down Expand Up @@ -1152,10 +1153,7 @@ def dict_keys_repr(const_keys, *, local) -> str:
GLOBAL_KEY_PREFIX = "__dict_key"


from torch._subclasses import ( # noqa: F401
FakeTensorMode,
UnsupportedFakeTensorException,
)
from torch._subclasses import UnsupportedFakeTensorException # noqa: F401


def wrap_fake_exception(fn):
Expand Down

0 comments on commit 4d23fcd

Please sign in to comment.