Force fixed version into sdist

Author: mgorny

.github/workflows/create_release.yml

@@ -46,7 +46,8 @@ jobs:
           if [[ ${tag_or_branch} == v* ]]; then
             # strip trailing v from tag name
             tag_or_branch="${tag_or_branch#v}"
-            echo "TRITON_OVERRIDE_VERSION=${tag_or_branch}" >> "$GITHUB_ENV"
+            # important: version must be fixed in setup.py
+            sed -i -e "s:^TRITON_VERSION = .*:TRITON_VERSION = '${tag_or_branch}':" setup.py || exit 1
           fi
           echo "RELEASE_NAME=triton-$tag_or_branch" >> "$GITHUB_ENV"
       - name: Create source distribution

setup.py

@@ -748,10 +748,12 @@ def get_git_version_suffix():
 # will cause errors
 Path(__file__).parent.joinpath("python", "triton", "_C").mkdir(exist_ok=True)
 
+# keep it separate for easy substitution
+TRITON_VERSION = "3.3.0" + get_git_version_suffix() + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", "")
+
 setup(
     name=os.environ.get("TRITON_WHEEL_NAME", "triton"),
-    version=os.environ.get("TRITON_OVERRIDE_VERSION",
-                           "3.3.0" + get_git_version_suffix() + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", "")),
+    version=TRITON_VERSION,
     author="Philippe Tillet",
     author_email="phil@openai.com",
     description="A language and compiler for custom Deep Learning operations",