xref: /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/fake_tensor.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
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