[Add] paddle samples and testing tools

Author: hxzd5568

graph_net/paddle/__init__.py

@@ -0,0 +1,7 @@
+"""
+GraphNet Paddle Implementation
+"""
+
+from .samples_util import get_default_samples_directory
+
+__all__ = ["get_default_samples_directory"]

graph_net/paddle/check_redundant_incrementally.py

@@ -0,0 +1,84 @@
+from . import utils
+import argparse
+import importlib.util
+import inspect
+from pathlib import Path
+from typing import Type, Any
+import sys
+import os
+import os.path
+from dataclasses import dataclass
+from contextlib import contextmanager
+import time
+import glob
+
+
+def get_recursively_model_pathes(root_dir):
+    for sub_dir in _get_recursively_model_pathes(root_dir):
+        yield os.path.realpath(sub_dir)
+
+
+def _get_recursively_model_pathes(root_dir):
+    if is_single_model_dir(root_dir):
+        yield root_dir
+        return
+    for sub_dir in get_immediate_subdirectory_paths(root_dir):
+        if is_single_model_dir(sub_dir):
+            yield sub_dir
+        else:
+            yield from get_recursively_model_pathes(sub_dir)
+
+
+def get_immediate_subdirectory_paths(parent_dir):
+    return [
+        sub_dir
+        for name in os.listdir(parent_dir)
+        for sub_dir in [os.path.join(parent_dir, name)]
+        if os.path.isdir(sub_dir)
+    ]
+
+
+def is_single_model_dir(model_dir):
+    return os.path.isfile(f"{model_dir}/graph_net.json")
+
+
+def main(args):
+    assert os.path.isdir(args.model_path)
+    assert os.path.isdir(args.graph_net_samples_path)
+    current_model_graph_hash_pathes = set(
+        graph_hash_path
+        for model_path in get_recursively_model_pathes(args.model_path)
+        for graph_hash_path in [f"{model_path}/graph_hash.txt"]
+    )
+    graph_hash2graph_net_model_path = {
+        graph_hash: graph_hash_path
+        for model_path in get_recursively_model_pathes(args.graph_net_samples_path)
+        for graph_hash_path in [f"{model_path}/graph_hash.txt"]
+        if os.path.isfile(graph_hash_path)
+        if graph_hash_path not in current_model_graph_hash_pathes
+        for graph_hash in [open(graph_hash_path).read()]
+    }
+    for current_model_graph_hash_path in current_model_graph_hash_pathes:
+        graph_hash = open(current_model_graph_hash_path).read()
+        assert (
+            graph_hash not in graph_hash2graph_net_model_path
+        ), f"Redundant models detected. old-model-path:{current_model_graph_hash_path}, new-model-path:{graph_hash2graph_net_model_path[graph_hash]}."
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Test compiler performance.")
+    parser.add_argument(
+        "--model-path",
+        type=str,
+        required=True,
+        help="Path to model file(s), each subdirectory containing graph_net.json will be regarded as a model",
+    )
+    parser.add_argument(
+        "--graph-net-samples-path",
+        type=str,
+        required=False,
+        default="default",
+        help="Path to GraphNet samples",
+    )
+    args = parser.parse_args()
+    main(args=args)

graph_net/paddle/remove_redundant_incrementally.py

@@ -0,0 +1,87 @@
+from . import utils
+import argparse
+import importlib.util
+import inspect
+from pathlib import Path
+from typing import Type, Any
+import sys
+import os
+import os.path
+from dataclasses import dataclass
+from contextlib import contextmanager
+import time
+import glob
+import shutil
+
+
+def get_recursively_model_pathes(root_dir):
+    for sub_dir in _get_recursively_model_pathes(root_dir):
+        yield os.path.realpath(sub_dir)
+
+
+def _get_recursively_model_pathes(root_dir):
+    if is_single_model_dir(root_dir):
+        yield root_dir
+        return
+    for sub_dir in get_immediate_subdirectory_paths(root_dir):
+        if is_single_model_dir(sub_dir):
+            yield sub_dir
+        else:
+            yield from get_recursively_model_pathes(sub_dir)
+
+
+def get_immediate_subdirectory_paths(parent_dir):
+    return [
+        sub_dir
+        for name in os.listdir(parent_dir)
+        for sub_dir in [os.path.join(parent_dir, name)]
+        if os.path.isdir(sub_dir)
+    ]
+
+
+def is_single_model_dir(model_dir):
+    return os.path.isfile(f"{model_dir}/graph_net.json")
+
+
+def main(args):
+    assert os.path.isdir(args.model_path)
+    assert os.path.isdir(args.graph_net_samples_path)
+    current_model_graph_hash_pathes = set(
+        graph_hash_path
+        for model_path in get_recursively_model_pathes(args.model_path)
+        for graph_hash_path in [f"{model_path}/graph_hash.txt"]
+    )
+    graph_hash2graph_net_model_path = {
+        graph_hash: graph_hash_path
+        for model_path in get_recursively_model_pathes(args.graph_net_samples_path)
+        for graph_hash_path in [f"{model_path}/graph_hash.txt"]
+        if os.path.isfile(graph_hash_path)
+        if graph_hash_path not in current_model_graph_hash_pathes
+        for graph_hash in [open(graph_hash_path).read()]
+    }
+    for current_model_graph_hash_path in current_model_graph_hash_pathes:
+        graph_hash = open(current_model_graph_hash_path).read()
+        if graph_hash not in graph_hash2graph_net_model_path:
+            continue
+        directory = os.path.dirname(current_model_graph_hash_path)
+        shutil.rmtree(directory)
+        os.makedirs(directory, exist_ok=True)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Test compiler performance.")
+    parser.add_argument(
+        "--model-path",
+        type=str,
+        required=True,
+        help="Path to model file(s), each subdirectory containing graph_net.json will be regarded as a model",
+    )
+    parser.add_argument(
+        "--graph-net-samples-path",
+        type=str,
+        required=False,
+        default="default",
+        help="Path to GraphNet samples",
+    )
+    args = parser.parse_args()
+    main(args=args)

graph_net/paddle/samples_util.py

@@ -0,0 +1,6 @@
+import graph_net
+import os
+
+
+def get_default_samples_directory():
+    return f"{os.path.dirname(graph_net.__file__)}/../samples"