Enable CXX_ABI=1 build flag for special wheel build

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.47k stars 473 forks source link

Enable CXX_ABI=1 build flag for special wheel build #8306

Open zpcore opened 5 hours ago

zpcore commented 5 hours ago

🚀 Feature

Enable CXX_ABI=1 flag in nightly build and 2.6 release.

Motivation

We have seen tracing time performance improvement when build torch_xla with CXX_ABI=1 flag on thanks to @tengyifei . We want to provide special build wheel in our daily build pipeline.

This issue is to track the progress.

zpcore commented 5 hours ago

@ManfeiBai @bhavya01 @miladm

miladm commented 5 hours ago

Justification: we observe significantly better perf with this flag on several models including SD2