Skip to content

Move inputgen and specdb under facto #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions calibrator/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,11 @@

import torch
from executorch.exir.dialects.edge.op.api import get_callable, to_variant
from inputgen.argtuple.engine import MetaArgTupleEngine
from inputgen.argtuple.gen import ArgumentTupleGenerator
from inputgen.argument.engine import MetaArg
from inputgen.specs.model import Spec
from specdb.db import SpecDictDB
from facto.inputgen.argtuple.engine import MetaArgTupleEngine
from facto.inputgen.argtuple.gen import ArgumentTupleGenerator
from facto.inputgen.argument.engine import MetaArg
from facto.inputgen.specs.model import Spec
from facto.specdb.db import SpecDictDB
from torch._ops import OpOverload
from torchgen.model import SchemaKind

Expand Down
4 changes: 2 additions & 2 deletions examples/example.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
# LICENSE file in the root directory of this source tree.

import torch
from inputgen.argtuple.gen import ArgumentTupleGenerator
from specdb.db import SpecDictDB
from facto.inputgen.argtuple.gen import ArgumentTupleGenerator
from facto.specdb.db import SpecDictDB


def pretty_print_add_args(posargs, inkwargs, outargs):
Expand Down
4 changes: 2 additions & 2 deletions examples/minimal_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
# LICENSE file in the root directory of this source tree.

import torch
from inputgen.argtuple.gen import ArgumentTupleGenerator
from specdb.db import SpecDictDB
from facto.inputgen.argtuple.gen import ArgumentTupleGenerator
from facto.specdb.db import SpecDictDB


def main():
Expand Down
6 changes: 3 additions & 3 deletions examples/random_seed.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@
# LICENSE file in the root directory of this source tree.

import torch
from inputgen.argtuple.gen import ArgumentTupleGenerator
from inputgen.utils.random_manager import random_manager
from specdb.db import SpecDictDB
from facto.inputgen.argtuple.gen import ArgumentTupleGenerator
from facto.inputgen.utils.random_manager import random_manager
from facto.specdb.db import SpecDictDB


def main():
Expand Down
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from inputgen.argument.engine import MetaArgEngine
from inputgen.argument.gen import ArgumentGenerator
from inputgen.attribute.model import Attribute
from inputgen.specs.model import Spec
from facto.inputgen.argument.engine import MetaArgEngine
from facto.inputgen.argument.gen import ArgumentGenerator
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.specs.model import Spec


def reverse_topological_sort(graph):
Expand Down
8 changes: 4 additions & 4 deletions inputgen/argtuple/gen.py → facto/inputgen/argtuple/gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@
from collections import OrderedDict
from typing import Any, Generator, List, Tuple

from inputgen.argtuple.engine import MetaArgTupleEngine
from inputgen.argument.engine import MetaArg
from inputgen.argument.gen import ArgumentGenerator
from inputgen.specs.model import Spec
from facto.inputgen.argtuple.engine import MetaArgTupleEngine
from facto.inputgen.argument.engine import MetaArg
from facto.inputgen.argument.gen import ArgumentGenerator
from facto.inputgen.specs.model import Spec


class ArgumentTupleGenerator:
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,13 @@
from typing import Any, List, Optional, Tuple, Union

import torch
from inputgen.argument.type import ArgType
from inputgen.attribute.engine import AttributeEngine
from inputgen.attribute.model import Attribute
from inputgen.attribute.solve import AttributeSolver
from inputgen.specs.model import Constraint, ConstraintSuffix
from inputgen.utils.random_manager import random_manager as rm
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.engine import AttributeEngine
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.attribute.solve import AttributeSolver
from facto.inputgen.specs.model import Constraint, ConstraintSuffix
from facto.inputgen.utils.random_manager import random_manager as rm
from facto.inputgen.variable.type import ScalarDtype


class StructuralEngine:
Expand Down
8 changes: 4 additions & 4 deletions inputgen/argument/gen.py → facto/inputgen/argument/gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,10 @@
from typing import Optional, Tuple

import torch
from inputgen.argument.engine import MetaArg
from inputgen.utils.random_manager import random_manager
from inputgen.variable.gen import VariableGenerator
from inputgen.variable.space import VariableSpace
from facto.inputgen.argument.engine import MetaArg
from facto.inputgen.utils.random_manager import random_manager
from facto.inputgen.variable.gen import VariableGenerator
from facto.inputgen.variable.space import VariableSpace
from torch.testing._internal.common_dtype import floating_types, integral_types


Expand Down
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,12 @@
import copy
from typing import List, Optional

from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from inputgen.attribute.solve import AttributeSolver
from inputgen.specs.model import Constraint
from inputgen.variable.gen import VariableGenerator
from inputgen.variable.type import ScalarDtype, sort_values_of_type
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.attribute.solve import AttributeSolver
from facto.inputgen.specs.model import Constraint
from facto.inputgen.variable.gen import VariableGenerator
from facto.inputgen.variable.type import ScalarDtype, sort_values_of_type


class AttributeEngine(AttributeSolver):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
from typing import List, Optional, Tuple

import torch
from inputgen.argument.type import ArgType
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.variable.type import ScalarDtype


class Attribute(str, Enum):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@

from typing import Any, List, Optional

from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from inputgen.specs.model import Constraint, ConstraintSuffix
from inputgen.variable.solve import SolvableVariable
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.specs.model import Constraint, ConstraintSuffix
from facto.inputgen.variable.solve import SolvableVariable
from facto.inputgen.variable.type import ScalarDtype


class AttributeSolver:
Expand Down
File renamed without changes.
File renamed without changes.
4 changes: 2 additions & 2 deletions inputgen/specs/model.py → facto/inputgen/specs/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@
from enum import Enum
from typing import Callable, List, Optional

from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute


class ConstraintSuffix(str, Enum):
Expand Down
File renamed without changes.
File renamed without changes.
File renamed without changes.
10 changes: 5 additions & 5 deletions inputgen/variable/gen.py → facto/inputgen/variable/gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,11 @@
import math
from typing import Any, List, Optional, Set, Union

from inputgen.utils.random_manager import random_manager as rm
from inputgen.variable.constants import BOUND_ON_INF, INT64_MAX, INT64_MIN
from inputgen.variable.space import Interval, Intervals, VariableSpace
from inputgen.variable.type import sort_values_of_type
from inputgen.variable.utils import nextdown, nextup
from facto.inputgen.utils.random_manager import random_manager as rm
from facto.inputgen.variable.constants import BOUND_ON_INF, INT64_MAX, INT64_MIN
from facto.inputgen.variable.space import Interval, Intervals, VariableSpace
from facto.inputgen.variable.type import sort_values_of_type
from facto.inputgen.variable.utils import nextdown, nextup


def gen_min_float_from_interval(r: Interval) -> Optional[float]:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
import math
from typing import Any, List, Union

from inputgen.variable.space import Discrete, VariableSpace
from inputgen.variable.type import convert_to_vtype, invalid_vtype, is_integer
from facto.inputgen.variable.space import Discrete, VariableSpace
from facto.inputgen.variable.type import convert_to_vtype, invalid_vtype, is_integer


class SolvableVariable:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
import math
from typing import Any, Callable, List, Optional, Union

from inputgen.variable.constants import INT64_MAX, INT64_MIN
from inputgen.variable.type import (
from facto.inputgen.variable.constants import INT64_MAX, INT64_MIN
from facto.inputgen.variable.type import (
invalid_vtype,
is_integer,
ScalarDtype,
Expand Down
File renamed without changes.
File renamed without changes.
12 changes: 6 additions & 6 deletions specdb/db.py → facto/specdb/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,19 +6,19 @@

import math

import specdb.dtypes as dt
import specdb.function as fn
import facto.specdb.dtypes as dt
import facto.specdb.function as fn
import torch
from inputgen.argument.type import ArgType
from inputgen.specs.model import (
from facto.inputgen.argument.type import ArgType
from facto.inputgen.specs.model import (
ConstraintProducer as cp,
InKwArg,
InPosArg,
OutArg,
Spec,
)
from inputgen.variable.type import ScalarDtype
from specdb.default import (
from facto.inputgen.variable.type import ScalarDtype
from facto.specdb.default import (
DimDefault,
DimListDefault,
IndexDefault,
Expand Down
4 changes: 2 additions & 2 deletions specdb/default.py → facto/specdb/default.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

import specdb.function as fn
from inputgen.specs.model import ConstraintProducer as cp
import facto.specdb.function as fn
from facto.inputgen.specs.model import ConstraintProducer as cp


DimDefault = [
Expand Down
File renamed without changes.
4 changes: 2 additions & 2 deletions specdb/function.py → facto/specdb/function.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
import random

import torch
from inputgen.variable.type import ScalarDtype
from inputgen.variable.utils import nextdown, nextup
from facto.inputgen.variable.type import ScalarDtype
from facto.inputgen.variable.utils import nextdown, nextup


def safe_ix(array, ix, default=0):
Expand Down
6 changes: 3 additions & 3 deletions test/inputgen/test_argtuple_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@

import unittest

from inputgen.argtuple.engine import MetaArgTupleEngine
from inputgen.argument.type import ArgType
from inputgen.specs.model import ConstraintProducer as cp, InPosArg, Spec
from facto.inputgen.argtuple.engine import MetaArgTupleEngine
from facto.inputgen.argument.type import ArgType
from facto.inputgen.specs.model import ConstraintProducer as cp, InPosArg, Spec


class TestMetaArgTupleEngine(unittest.TestCase):
Expand Down
6 changes: 3 additions & 3 deletions test/inputgen/test_argtuple_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
import unittest

import torch
from inputgen.argtuple.gen import ArgumentTupleGenerator
from inputgen.argument.type import ArgType
from inputgen.specs.model import ConstraintProducer as cp, InPosArg, Spec
from facto.inputgen.argtuple.gen import ArgumentTupleGenerator
from facto.inputgen.argument.type import ArgType
from facto.inputgen.specs.model import ConstraintProducer as cp, InPosArg, Spec


class TestArgumentTupleGenerator(unittest.TestCase):
Expand Down
6 changes: 3 additions & 3 deletions test/inputgen/test_argument_attributes.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
import unittest

import torch
from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.variable.type import ScalarDtype


class TestAttribute(unittest.TestCase):
Expand Down
8 changes: 4 additions & 4 deletions test/inputgen/test_argument_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@
import unittest

import torch
from inputgen.argument.engine import MetaArg
from inputgen.argument.gen import ArgumentGenerator, TensorGenerator
from inputgen.argument.type import ArgType
from inputgen.variable.solve import SolvableVariable
from facto.inputgen.argument.engine import MetaArg
from facto.inputgen.argument.gen import ArgumentGenerator, TensorGenerator
from facto.inputgen.argument.type import ArgType
from facto.inputgen.variable.solve import SolvableVariable


class TestTensorGenerator(unittest.TestCase):
Expand Down
2 changes: 1 addition & 1 deletion test/inputgen/test_argument_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

import unittest

from inputgen.argument.type import ArgType
from facto.inputgen.argument.type import ArgType


class TestArgType(unittest.TestCase):
Expand Down
10 changes: 5 additions & 5 deletions test/inputgen/test_attribute_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,11 @@
import unittest

import torch
from inputgen.argument.type import ArgType
from inputgen.attribute.engine import AttributeEngine
from inputgen.attribute.model import Attribute
from inputgen.specs.model import ConstraintProducer as cp
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.engine import AttributeEngine
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.specs.model import ConstraintProducer as cp
from facto.inputgen.variable.type import ScalarDtype


class TestAttributeEngine(unittest.TestCase):
Expand Down
10 changes: 5 additions & 5 deletions test/inputgen/test_attribute_solver.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@

import unittest

from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from inputgen.attribute.solve import AttributeSolver
from inputgen.specs.model import ConstraintProducer as cp
from inputgen.variable.type import ScalarDtype
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.attribute.solve import AttributeSolver
from facto.inputgen.specs.model import ConstraintProducer as cp
from facto.inputgen.variable.type import ScalarDtype


class TestAttributeSolver(unittest.TestCase):
Expand Down
8 changes: 6 additions & 2 deletions test/inputgen/test_constraints.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,12 @@

import unittest

from inputgen.attribute.model import Attribute
from inputgen.specs.model import Constraint, ConstraintProducer as cp, ConstraintSuffix
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.specs.model import (
Constraint,
ConstraintProducer as cp,
ConstraintSuffix,
)


class TestConstraint(unittest.TestCase):
Expand Down
10 changes: 5 additions & 5 deletions test/inputgen/test_meta_arg_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@

import unittest

from inputgen.argument.engine import MetaArgEngine
from inputgen.argument.type import ArgType
from inputgen.attribute.model import Attribute
from inputgen.specs.model import ConstraintProducer as cp
from inputgen.variable.type import SUPPORTED_TENSOR_DTYPES
from facto.inputgen.argument.engine import MetaArgEngine
from facto.inputgen.argument.type import ArgType
from facto.inputgen.attribute.model import Attribute
from facto.inputgen.specs.model import ConstraintProducer as cp
from facto.inputgen.variable.type import SUPPORTED_TENSOR_DTYPES


class TestMetaArgEngine(unittest.TestCase):
Expand Down
Loading
Loading