Rename package for tefl

Signed-off-by: wenjh <wenjh@sugon.com>

build_tools/build_ext.py

@@ -148,7 +148,7 @@ def get_build_ext(
                 if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))):
                     target_dir = install_dir / "transformer_engine" / lib_dir
                 else:
-                    target_dir = install_dir / "transformer_engine_fl_hygon" / lib_dir
+                    target_dir = install_dir / "transformer_engine_hygon" / lib_dir
                 target_dir.mkdir(exist_ok=True, parents=True)
 
                 for ext in Path(self.build_lib).glob("*.so"):

build_tools/pytorch.py

@@ -155,7 +155,7 @@ def setup_pytorch_extension(
 
     if rocm_build():
         return CUDAExtension(
-            name="transformer_engine_torch" if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))) else "transformer_engine_fl_torch_hygon",
+            name="transformer_engine_torch" if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))) else "transformer_engine_torch_hygon",
             sources=[str(src) for src in sources],
             include_dirs=[str(inc) for inc in include_dirs],
             extra_compile_args={

setup.py

@@ -52,12 +52,12 @@ else:
 
 if bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))):
     common_dir = current_file_path / "transformer_engine" / "common"
-    common_copy = current_file_path / "transformer_engine_fl_hygon" / "common"
+    common_copy = current_file_path / "transformer_engine_hygon" / "common"
     if common_copy.exists():
         shutil.rmtree(common_copy)
     shutil.copytree(common_dir, common_copy)
     csrc_dir = current_file_path / "transformer_engine" / "pytorch" / "csrc"
-    csrc_copy = current_file_path / "transformer_engine_fl_hygon" / "pytorch" / "csrc"
+    csrc_copy = current_file_path / "transformer_engine_hygon" / "pytorch" / "csrc"
     if csrc_copy.exists():
         shutil.rmtree(csrc_copy)
     shutil.copytree(csrc_dir, csrc_copy)
@@ -128,9 +128,9 @@ def setup_common_extension() -> CMakeExtension:
     if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))):
         cmake_path = root_path / Path("transformer_engine/common")
     else:
-        cmake_path = root_path / Path("transformer_engine_fl_hygon/common")
+        cmake_path = root_path / Path("transformer_engine_hygon/common")
     return CMakeExtension(
-        name="transformer_engine" if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))) else "transformer_engine_fl_hygon",
+        name="transformer_engine" if not bool(int(os.getenv("TEFL_HYGON_BACKEND", "0"))) else "transformer_engine_hygon",
         cmake_path=cmake_path,
         cmake_flags=cmake_flags,
     )
@@ -265,9 +265,9 @@ if __name__ == "__main__":
                 else:
                     ext_modules.append(
                         setup_pytorch_extension(
-                            "transformer_engine_fl_hygon/pytorch/csrc",
-                            current_file_path / "transformer_engine_fl_hygon" / "pytorch" / "csrc",
-                            current_file_path / "transformer_engine_fl_hygon",
+                            "transformer_engine_hygon/pytorch/csrc",
+                            current_file_path / "transformer_engine_hygon" / "pytorch" / "csrc",
+                            current_file_path / "transformer_engine_hygon",
                         )
                     )
             if "jax" in frameworks:
@@ -309,12 +309,12 @@ if __name__ == "__main__":
     else:
         # Configure package of hygon backend for TransformerEngine-FL
         setuptools.setup(
-            name="transformer_engine_fl_hygon",
+            name="transformer_engine_hygon",
             version=__version__,
             packages=setuptools.find_packages(
                 include=[
-                    "transformer_engine_fl_hygon",
-                    "transformer_engine_fl_hygon.*",
+                    "transformer_engine_hygon",
+                    "transformer_engine_hygon.*",
                 ],
             ),
             extras_require=extras_require,