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

[nativert] get rid of sigmoid naming (#157134)

Browse Source 
Summary: att

Test Plan:
ci

Rollback Plan:

Differential Revision: D77451215

Pull Request resolved: https://github.com/pytorch/pytorch/pull/157134
Approved by: https://github.com/zhxchen17, https://github.com/jingsh
This commit is contained in:
Dylan Maloy
2025-06-27 22:41:48 +00:00
committed by PyTorch MergeBot
parent d283fc79b1
commit cd1a924dba
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nativert/executor/memory/AliasAnalyzer.cpp
View File

@ -148,7 +148,7 @@ void AliasAnalyzer::log_state() const {
std::cout << [&]() -> std::string {
std::ostringstream ss;
ss << "[sigmoid layout planner] AliasAnalyzer ran....\n";
ss << "[nativert layout planner] AliasAnalyzer ran....\n";
ss << "lifetimes:\n";
for (const auto& [v, lifetime] : lifetimes_) {

4
torch/nativert/kernels/PrimKernelRegistry.cpp
View File

@ -35,7 +35,7 @@ class OpKernel_prim_listpack : public OpKernel {
}
void computeInternal(ExecutionFrame& executionFrame) const override final {
RECORD_USER_SCOPE("sigmoid::OpKernel_prim_listpack");
RECORD_USER_SCOPE("nativert::OpKernel_prim_listpack");
c10::List<c10::IValue> list(type_);
list.reserve(numInputs());
for (size_t i = 0; i < numInputs(); ++i) {
@ -60,7 +60,7 @@ C10_REGISTER_TYPED_CLASS(
OpKernel_prim_listpack);
REGISTER_PRIM_KERNEL("prim.ListUnpack", prim_listunpack, {
RECORD_USER_SCOPE("sigmoid::OpKernel_prim_listunpack");
RECORD_USER_SCOPE("nativert::OpKernel_prim_listunpack");
auto inputListRef = KernelInput(0).toListRef();
for (const auto& [i, ivalue] : c10::enumerate(inputListRef)) {
KernelOutput(i) = ivalue;