From 291abfe6e86df89698b34390ed6a077fa076bf33 Mon Sep 17 00:00:00 2001 From: Daniel Li Date: Fri, 27 Dec 2024 15:36:28 +0800 Subject: [PATCH] Add pytorch binary build support for l4t35.6 --- tools/generate_torch_version.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/generate_torch_version.py b/tools/generate_torch_version.py index 8c5f950141e00..ed4f5b5dd6f22 100644 --- 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: