Merge pull request #24 from mlcommons/deprecate-flexflow

et_converter: Deprecate flexfow2chakra converter
mlcommons · Feb 8, 2024 · e053939 · e053939
2 parents 13d9961 + dd37d5a
commit e053939
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 233 deletions.
diff --git a/et_converter/et_converter.py b/et_converter/et_converter.py
@@ -7,7 +7,6 @@
 
 from logging import FileHandler
 from .text2chakra_converter import Text2ChakraConverter
-from .flexflow2chakra_converter import FlexFlow2ChakraConverter
 from .pytorch2chakra_converter import PyTorch2ChakraConverter
 
 def get_logger(log_filename: str) -> logging.Logger:
@@ -69,12 +68,6 @@ def main() -> None:
         default=None,
         required="Text" in sys.argv,
         help="Number of training passes")
-    parser.add_argument(
-        "--npu_frequency",
-        type=int,
-        default=None,
-        required="FlexFlow" in sys.argv,
-        help="NPU frequency in MHz")
     parser.add_argument(
         "--log_filename",
         type=str,
@@ -95,14 +88,6 @@ def main() -> None:
                 args.num_passes,
                 logger)
             converter.convert()
-        elif args.input_type == "FlexFlow":
-            converter = FlexFlow2ChakraConverter(
-                args.input_filename,
-                args.output_filename,
-                args.num_dims,
-                args.npu_frequency,
-                logger)
-            converter.convert()
         elif args.input_type == "PyTorch":
             converter = PyTorch2ChakraConverter(
                 args.input_filename,

diff --git a/et_converter/flexflow2chakra_converter.py b/et_converter/flexflow2chakra_converter.py