reorder symbol names

Author: lixinqi

graph_net/constraint_util.py

@@ -143,12 +143,13 @@ def symbolize_data_input_dims(
     Returns new DynamicDimConstraints if success.
     Returns None if no symbolicable dim .
     """
-    unqiue_dims = set()
+    unqiue_dims = []
 
     def dumpy_filter_fn(input_name, input_idx, axis, dim):
         if is_data_input(input_name):
             print("data_input", input_name, input_idx, axis, dim)
-            unqiue_dims.add(dim)
+            if dim not in unqiue_dims:
+                unqiue_dims.append(dim)
         # No symbolization because of returning True
         return False
 

graph_net/tensor_meta.py

@@ -5,6 +5,7 @@
 
 @dataclass
 class TensorMeta:
+    record_class_name: str
     name: str
     original_name: str | None
     shape: list[int]
@@ -20,6 +21,7 @@ class TensorMeta:
     def unserialize_from_py_file(cls, file_path) -> list["TensorMeta"]:
         return [
             TensorMeta(
+                record_class_name=attrs.get("record_class_name"),
                 name=attrs.get("name"),
                 original_name=attrs.get("original_name", None),
                 shape=attrs.get("shape", []),
@@ -37,11 +39,13 @@ def unserialize_from_py_file(cls, file_path) -> list["TensorMeta"]:
 
     @classmethod
     def _convert_cls_to_attrs(cls, tensor_meta_cls):
-        return {
+        attrs = {
             k: v
             for k, v in tensor_meta_cls.__dict__.items()
             if not k.startswith("__") and not callable(v)
         }
+        attrs["record_class_name"] = tensor_meta_cls.__name__
+        return attrs
 
     @classmethod
     def _get_classes(cls, file_path, name="unamed"):
@@ -50,10 +54,9 @@ def _get_classes(cls, file_path, name="unamed"):
         spec.loader.exec_module(unnamed)
         yield from inspect.getmembers(unnamed, inspect.isclass)
 
-    def serialize_to_py_str(self, cls_name_prefix: str = "example_input") -> str:
-        uid = f"{cls_name_prefix}_tensor_meta_{self.name}"
+    def serialize_to_py_str(self) -> str:
         lines = [
-            (f"class {uid}:"),
+            (f"class {self.record_class_name}:"),
             (f'\tname = "{self.name}"'),
             (f"\tshape = {self.shape}"),
             (f'\tdtype = "{self.dtype}"'),

samples/timm/resnet18/input_tensor_constraints.py

@@ -5,7 +5,7 @@
 
 dynamic_dim_constraint_symbols = [S0, S1]
 
-dynamic_dim_constraint_symbol2example_value = {S0: 224, S1: 1}
+dynamic_dim_constraint_symbol2example_value = {S0: 1, S1: 224}
 
 dynamic_dim_constraint_relations = []
 
@@ -205,7 +205,7 @@
         [512, 512, 3, 3],
         "L_self_modules_layer4_modules_1_modules_conv2_parameters_weight_",
     ),
-    ([S1, 3, S0, S0], "L_x_"),
+    ([S0, 3, S1, S1], "L_x_"),
     ([], "s1"),
 ]