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] Move call_torchbind_kernel (#156571)

Browse Source 
Summary: Move call_torchbind_kernel target from internal sigmoid to pytorch

Test Plan:
Test Internally:

buck2 test mode/dev-nosan caffe2/test/cpp/nativert:op_kernel_test
buck build //sigmoid/core/kernels:kernel_factory
and all  sandcastle tests

Rollback Plan:

Differential Revision: D77118592

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156571
Approved by: https://github.com/zhxchen17
This commit is contained in:
Jeddie Ji
2025-06-24 15:24:06 +00:00
committed by PyTorch MergeBot
parent 795a6a0aff
commit 4c59edf0c5
3 changed files with 78 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
build_variables.bzl
View File

@ -613,6 +613,7 @@ libtorch_nativert_sources = [
"torch/nativert/kernels/HigherOrderKernel.cpp",
"torch/nativert/executor/memory/GreedyBySize.cpp",
"torch/nativert/executor/memory/Bump.cpp",
"torch/nativert/kernels/CallTorchBindKernel.cpp",
]
torch_mobile_tracer_sources = [