Add pytorch binary build support for l4t35.6

SnapDragonfly · Jan 6, 2025 · 291abfe · 291abfe
1 parent a8d6afb
commit 291abfe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/generate_torch_version.py b/tools/generate_torch_version.py
@@ -55,6 +55,8 @@ def get_torch_version(sha: str | None = None) -> str:
         assert os.getenv("PYTORCH_BUILD_NUMBER") is not None
         build_number = int(os.getenv("PYTORCH_BUILD_NUMBER", ""))
         version = os.getenv("PYTORCH_BUILD_VERSION", "")
+        l4t_ver = os.getenv("L4T_BUILD_VERSION", "")
+        version += "+l4t" + l4t_ver
         if build_number > 1:
             version += ".post" + str(build_number)
     elif sha != UNKNOWN: