1# mypy: ignore-errors 2 3import torch._subclasses 4 5 6def is_builtin(op): 7 return op.namespace in ('aten', 'prims', 'prim') 8 9 10def fake_check(op, args, kwargs): 11 with torch._subclasses.CrossRefFakeMode(ignore_op_fn=is_builtin): 12 op(*args, **kwargs) 13