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

[nativert] Move KernelFactory to PyTorch core (#156913)

Browse Source 
Summary: Kernel factory handles the kernel nodes initializations and different type of kernels executions.

Test Plan:
CI

Rollback Plan:

Differential Revision: D77346836

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156913
Approved by: https://github.com/zhxchen17
This commit is contained in:
Sheng Qin
2025-06-28 06:34:21 +00:00
committed by PyTorch MergeBot
parent 51eb8e8f84
commit 88c6199db0
3 changed files with 360 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
build_variables.bzl
View File

@ -617,6 +617,7 @@ libtorch_nativert_sources = [
"torch/nativert/executor/memory/Bump.cpp",
"torch/nativert/executor/ParallelGraphExecutor.cpp",
"torch/nativert/kernels/CallTorchBindKernel.cpp",
"torch/nativert/kernels/KernelFactory.cpp",
"torch/nativert/kernels/PrimKernelRegistry.cpp",
"torch/nativert/executor/memory/DisjointStorageGroups.cpp",
"torch/nativert/executor/memory/AliasAnalyzer.cpp",