add annotation

frontend/Python/graph/__init__.py

@@ -1 +1,21 @@
+# ===- __init__.py -------------------------------------------------------
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ===---------------------------------------------------------------------------
+#
+# Init the packages in graph directory.
+#
+# ===---------------------------------------------------------------------------
+
 from .graph import Graph

frontend/Python/graph/graph.py

@@ -17,6 +17,7 @@
 # This is the graph level of the Buddy Compiler frontend.
 #
 # ===---------------------------------------------------------------------------
+
 import mlir.ir as ir
 from mlir.passmanager import *
 from mlir.execution_engine import *

frontend/Python/graph/graph_importer.py

@@ -1,11 +1,30 @@
+# ===- graph_importer.py -------------------------------------------------------
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ===---------------------------------------------------------------------------
+#
+# This is the graph importer for Buddy Compiler's graph.
+#
+# ===---------------------------------------------------------------------------
+
 import operator
 from typing import Any, List, Optional
 import functools
 
 import mlir.dialects.func as func
 import mlir.ir as ir
 import torch
-from torch._functorch.aot_autograd import aot_module_simplified
 
 
 class FXGraphImporter: