Add git SHA to wheel versions using versioningit

Signed-off-by: jbernloehr <[email protected]>

Author: janbernloehr

.github/workflows/build_whl_and_publish.yaml

@@ -35,6 +35,5 @@ jobs:
       test-infra-ref: main
       package-name: torchtitan
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: .github/scripts/update_version.sh
       trigger-event: ${{ github.event_name }}
       build-platform: 'python-build-package'

assets/version.txt

@@ -1,6 +1,6 @@
 ## Stable Releases
 Currently we follow a lightweight release process.
-- Update the version number in `assets/version.txt` with a PR. The version numbering should follow https://semver.org/.
+- Pick a version for the release. The version numbering should follow https://semver.org/.
   - E.g. for a pre-release `0.y.z`
     - if major features are added, increment `y`
     - if minor fixes are added, increment `z`

docs/release.md

@@ -42,22 +42,27 @@ dev = [
     "expecttest", # test_tokenizer
 ]
 
-[tool.setuptools.dynamic]
-version = {file = "assets/version.txt"}
-
 [tool.codespell]
 skip = ["*.json"]
 ignore-words-list = ["assertin", "datas" ,"indx", "inpt", "nd", "socio-economic"]
 
 # ---- Explicit project build information ---- #
 [build-system]
-requires = ["setuptools>=61.0"]
+requires = ["setuptools>=61.0", "versioningit>=3.3"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
 where = [""]
 include = ["torchtitan*"]
 
+[tool.versioningit.tag2version]
+rmprefix = "v"
+
+[tool.versioningit.format]
+distance = "{base_version}+git{rev}"
+dirty = "{base_version}+git{rev}.dirty"
+distance-dirty = "{base_version}+git{rev}.dirty"
+
 [tool.pytest.ini_options]
 addopts = ["--showlocals"]  # show local variables in tracebacks
 testpaths = ["tests"]