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 c10_kernel (#156208)

Browse Source 
Summary:
Torch Native Runtime RFC: https://github.com/pytorch/rfcs/pull/72

As part of the effort to open source TorchNativeRuntime (or what we call Sigmoid), we are moving the Pytree implementation to torch/:

fbcode/sigmoid/kernels -> fbcode/caffe2/torch/nativert/kernels

Test Plan:
```
buck run fbcode//mode/dev-nosan  //caffe2/test/cpp/nativert:c10_kernel_test
```

Differential Revision: D76825830

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156208
Approved by: https://github.com/zhxchen17
This commit is contained in:
Shangdi Yu
2025-06-19 17:36:23 +00:00
committed by PyTorch MergeBot
parent f402eed4d9
commit e4c9f6d9a2
5 changed files with 428 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
build_variables.bzl
View File

@ -606,6 +606,7 @@ libtorch_nativert_sources = [
"torch/nativert/executor/memory/FunctionSchema.cpp",
"torch/nativert/common/FileUtil.cpp",
"torch/nativert/detail/ITree.cpp",
"torch/nativert/kernels/C10Kernel.cpp",
]
torch_mobile_tracer_sources = [