Skip to content

Commit

Permalink
Revert "Disable torchrec/sparse from top-level Dynamo tracing (pytorc…
Browse files Browse the repository at this point in the history
…h#105733)"

This reverts commit 60d5efd.

Reverted pytorch#105733 on behalf of https://github.com/facebook-github-bot due to Diff reverted internally ([comment](pytorch#105733 (comment)))
  • Loading branch information
pytorchmergebot committed Jul 26, 2023
1 parent 884cd53 commit 6dd4b99
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions torch/_dynamo/skipfiles.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,12 +163,6 @@ def _module_dir(m: types.ModuleType):
FBCODE_SKIP_DIRS = {
"torchrec/distributed",
"torchrec/fb/distributed",
# JaggedTensor and KeyedJaggedTensor are fairly badly behaved, because
# they tend to contain extremely long lists of integers that vary from
# run to run. We are interested in tracing models that contain references
# to KJTs, but tracing an individual KeyedJaggedTensor is rarely
# profitable
"torchrec/sparse",
"caffe2/torch/fb/sparsenn/pooled_embeddings_modules.py",
}
FBCODE_SKIP_DIRS_RE = re.compile(f".*({'|'.join(map(re.escape, FBCODE_SKIP_DIRS))})")
Expand Down

0 comments on commit 6dd4b99

Please sign in to comment.