Rename *2chakra_converter to *_converter

mlcommons · May 9, 2024 · 38919e5 · 38919e5
1 parent 2a8640f
commit 38919e5
Show file tree

Hide file tree

Showing 5 changed files with 1,135 additions and 6 deletions.
diff --git a/src/converter/converter.py b/src/converter/converter.py
@@ -6,8 +6,8 @@
 import traceback
 from logging import FileHandler
 
-from .pytorch2chakra_converter import PyTorch2ChakraConverter
-from .text2chakra_converter import Text2ChakraConverter
+from .pytorch_converter import PyTorchConverter
+from .text_converter import TextConverter
 
 
 def get_logger(log_filename: str) -> logging.Logger:
@@ -52,12 +52,12 @@ def main() -> None:
 
     try:
         if args.input_type == "Text":
-            converter = Text2ChakraConverter(
+            converter = TextConverter(
                 args.input_filename, args.output_filename, args.num_npus, args.num_passes, logger
             )
             converter.convert()
         elif args.input_type == "PyTorch":
-            converter = PyTorch2ChakraConverter(args.input_filename, args.output_filename, logger)
+            converter = PyTorchConverter(args.input_filename, args.output_filename, logger)
             converter.convert()
         else:
             logger.error(f"{args.input_type} unsupported")

diff --git a/src/converter/pytorch2chakra_converter.py b/src/converter/pytorch2chakra_converter.py
@@ -27,7 +27,7 @@
 from .pytorch_node import PyTorchNode, PyTorchNodeType
 
 
-class PyTorch2ChakraConverter:
+class PyTorchConverter:
     """
     Converter class for transforming PyTorch execution traces into Chakra format.