commit 7af30ea54cc50faa220ef71e62faa564be2bbcd6
author Mu-Chu Lee <mlee8@meta.com> Mon Sep 25 14:49:15 2023 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Sep 25 23:03:06 2023 +0000
tree 727190de3c8ec1d8ed51bcc0f39af1c79ef67ee6
parent 6275f9165478de67b9e1548da48546078ea308c0

[AOTInductor] Bug fix for redefining symbol name (#110041)

Summary:
Bug fix for redefining symbol name.

Test Plan:
python benchmarks/dynamo/huggingface.py --bfloat16 --accuracy --inference --device cuda --export-aot-inductor --cold-start-latency --only OPTForCausalLM

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110041
Approved by: https://github.com/desertfire

torch/_inductor/codecache.py

diff --git a/torch/_inductor/codecache.py b/torch/_inductor/codecache.py
index 932c79c..cbfefaa 100644
--- a/torch/_inductor/codecache.py
+++ b/torch/_inductor/codecache.py
@@ -1102,7 +1102,7 @@
                         f"{objcopy_command} --redefine-sym _binary_{body}_start=_binary_constants_bin_start {consts_o}"
                     )
                     symbol_list.append(
-                        f"{objcopy_command} --redefine-sym _binary_{body}_start=_binary_constants_bin_size {consts_o}"
+                        f"{objcopy_command} --redefine-sym _binary_{body}_size=_binary_constants_bin_size {consts_o}"
                     )
                     symbol_list.append(
                         f"{objcopy_command} --redefine-sym _binary_{body}_end=_binary_constants_bin_end {consts_o}"