diff --git a/torch/_dynamo/skipfiles.py b/torch/_dynamo/skipfiles.py index 70bbcfa4472..aa16b405006 100644 --- a/torch/_dynamo/skipfiles.py +++ b/torch/_dynamo/skipfiles.py @@ -33,23 +33,6 @@ import torch import torch._export.constraints as _export_constraints import torch._inductor.test_operators - -try: - import torch._prims - - # isort: split - # TODO: Hack to unblock simultaneous landing changes. Fix after https://github.com/pytorch/pytorch/pull/81088 lands - import torch._prims.utils - import torch._prims.wrappers - import torch._refs - import torch._refs.nn - import torch._refs.nn.functional - import torch._refs.special - - HAS_PRIMS_REFS = True -except ImportError: - HAS_PRIMS_REFS = False - from . import comptime, config, external_utils """ @@ -139,16 +122,6 @@ FILENAME_ALLOWLIST |= { FILENAME_ALLOWLIST |= {torch.optim._functional.__file__} FILENAME_ALLOWLIST |= {_export_constraints.__file__} -if HAS_PRIMS_REFS: - FILENAME_ALLOWLIST |= { - torch._prims.__file__, - torch._prims.utils.__file__, - torch._prims.wrappers.__file__, - torch._refs.__file__, - torch._refs.special.__file__, - torch._refs.nn.functional.__file__, - } - SKIP_DIRS_RE = None