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] libtorch kernel registry (#157150)

Browse Source 
Summary: att

Test Plan:
ci

Rollback Plan:

Differential Revision: D77451703

Pull Request resolved: https://github.com/pytorch/pytorch/pull/157150
Approved by: https://github.com/georgiaphillips, https://github.com/henryoier
This commit is contained in:
dolpm
2025-07-16 12:36:51 +00:00
committed by PyTorch MergeBot
parent 55d888a616
commit 51a708ffc6
6 changed files with 1626 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
build_variables.bzl
View File

@ -625,6 +625,10 @@ libtorch_nativert_sources = [
"torch/nativert/executor/memory/AliasAnalyzer.cpp",
"torch/nativert/executor/memory/LayoutPlanner.cpp",
"torch/nativert/executor/memory/LayoutManager.cpp",
"torch/nativert/kernels/KernelRegistry.cpp",
"torch/nativert/kernels/NativeKernels.cpp",
"torch/nativert/kernels/GeneratedStaticDispatchKernels.cpp",
"torch/nativert/kernels/GeneratedNativeStaticDispatchKernels.cpp",
]
torch_mobile_tracer_sources = [