[BUILD] Update setup info for pypi (#163)

* [BUILD] Update setup info for pypi
* [BUILD] Uninstall triton when buiding flagtree locally

Author: zhzhcookie

python/setup.py

@@ -524,6 +524,7 @@ def add_links():
 class plugin_install(install):
 
     def run(self):
+        helper.uninstall_triton()
         add_links()
         install.run(self)
         helper.post_install()
@@ -532,6 +533,7 @@ def run(self):
 class plugin_develop(develop):
 
     def run(self):
+        helper.uninstall_triton()
         add_links()
         develop.run(self)
         helper.post_install()
@@ -624,13 +626,19 @@ def get_install_requires():
     packages, package_dir, package_data = helper.configure_cambricon_packages_and_data(
         packages, package_dir, package_data)
 
+readme_path = os.path.join(get_base_dir(), "README.md")
+with open(readme_path, "r", encoding="utf-8") as fh:
+    long_description = fh.read()
+
 setup(
-    name=os.environ.get("TRITON_WHEEL_NAME", "triton"),
-    version="3.1.0" + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", ""),
-    author="Philippe Tillet",
-    author_email="[email protected]",
-    description="A language and compiler for custom Deep Learning operations",
-    long_description="",
+    name=os.environ.get("FLAGTREE_WHEEL_NAME", "flagtree"),
+    version="0.3.0" + os.environ.get("FLAGTREE_WHEEL_VERSION_SUFFIX", ""),
+    author="FlagOS",
+    author_email="[email protected]",
+    description=
+    "A unified compiler supporting multiple AI chip backends for custom Deep Learning operations, which is forked from triton-lang/triton.",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
     packages=packages,
     package_dir=package_dir,
     entry_points=get_entry_points(),

python/setup_tools/setup_helper.py

@@ -360,6 +360,24 @@ def check_env(env_val):
     return os.environ.get(env_val, '') != ''
 
 
+def uninstall_triton():
+    is_bdist_wheel = any(cmd in sys.argv for cmd in ['bdist_wheel', 'egg_info', 'sdist'])
+    if is_bdist_wheel:
+        return
+    try:
+        import pkg_resources
+        import subprocess
+        try:
+            pkg_resources.get_distribution('triton')
+            print("Detected existing 'triton' package. Uninstalling to avoid conflicts...")
+            subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", "triton"])
+            print("Successfully uninstalled 'triton'.")
+        except pkg_resources.DistributionNotFound:
+            print("'triton' package not found, no need to uninstall.")
+    except Exception as e:
+        print(f"Warning: Failed to check/uninstall triton: {e}")
+
+
 offline_handler = utils.OfflineBuildManager()
 if offline_handler.is_offline:
     print("[INFO] Offline Build: Use offline build for triton origin toolkits")