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 1212import torch
1313from facto .inputgen .argtuple .gen import ArgumentTupleGenerator
1414from facto .inputgen .specs .model import ConstraintProducer as cp
15- from facto .inputgen .utils .random_manager import random_manager
1615from facto .inputgen .variable .type import ScalarDtype
1716from facto .specdb .db import SpecDictDB
17+ from functools import lru_cache
1818
1919# seed to generate identical cases every run to reproduce from bisect
20- random_manager .seed (1729 )
2120MAX_CASES = 50
2221
2322
@@ -103,7 +102,7 @@ def apply_scalar_contraints(op_name: str) -> list[ScalarDtype]:
103102 case _:
104103 return [ScalarDtype .float , ScalarDtype .int ]
105104
106-
105+ @ lru_cache ( maxsize = None )
107106def facto_testcase_gen (op_name : str ) -> List [Tuple [List [str ], OrderedDict [str , str ]]]:
108107 # minimal example to test add.Tensor using FACTO
109108 spec = SpecDictDB [op_name ]
You can’t perform that action at this time.
0 commit comments