File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 12
12
import torch
13
13
from facto .inputgen .argtuple .gen import ArgumentTupleGenerator
14
14
from facto .inputgen .specs .model import ConstraintProducer as cp
15
- from facto .inputgen .utils .random_manager import random_manager
16
15
from facto .inputgen .variable .type import ScalarDtype
17
16
from facto .specdb .db import SpecDictDB
17
+ from functools import lru_cache
18
18
19
19
# seed to generate identical cases every run to reproduce from bisect
20
- random_manager .seed (1729 )
21
20
MAX_CASES = 50
22
21
23
22
@@ -103,7 +102,7 @@ def apply_scalar_contraints(op_name: str) -> list[ScalarDtype]:
103
102
case _:
104
103
return [ScalarDtype .float , ScalarDtype .int ]
105
104
106
-
105
+ @ lru_cache ( maxsize = None )
107
106
def facto_testcase_gen (op_name : str ) -> List [Tuple [List [str ], OrderedDict [str , str ]]]:
108
107
# minimal example to test add.Tensor using FACTO
109
108
spec = SpecDictDB [op_name ]
You can’t perform that action at this time.
0 commit comments