frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[nativert] alias analyzer + layout planner/manager to pytorch core (#156897)

Browse Source 
Summary: att

Test Plan:
ci - unit tests still have some unresolved deps but will move them later.

Rollback Plan:

Differential Revision: D77320950

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156897
Approved by: https://github.com/zhxchen17
This commit is contained in:
dolpm
2025-06-27 03:01:22 +00:00
committed by PyTorch MergeBot
parent 382c6190c1
commit 7392470da4
7 changed files with 1002 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
build_variables.bzl
View File

@ -619,6 +619,9 @@ libtorch_nativert_sources = [
"torch/nativert/kernels/CallTorchBindKernel.cpp",
"torch/nativert/kernels/PrimKernelRegistry.cpp",
"torch/nativert/executor/memory/DisjointStorageGroups.cpp",
"torch/nativert/executor/memory/AliasAnalyzer.cpp",
"torch/nativert/executor/memory/LayoutPlanner.cpp",
"torch/nativert/executor/memory/LayoutManager.cpp",
]
torch_mobile_tracer_sources = [