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

[reland][ca] side-effect free inital trace: compiled_args (#148376)

Browse Source 
This reverts commit ea12fc8a9ff7da808e0b661ca07e9d4ce75d04bc.
Reland https://github.com/pytorch/pytorch/pull/147804, there was a bad import inserted by my linter.

Differential Revision: [D70582747](https://our.internmc.facebook.com/intern/diff/D70582747)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/148376
Approved by: https://github.com/jansel
This commit is contained in:
Simon Fan
2025-03-10 11:31:32 -07:00
committed by PyTorch MergeBot
parent 9fddbf3417
commit 457ff9b7ae
17 changed files with 104 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/csrc/autograd/utils/lambda_post_hook.h
View File

@ -27,7 +27,7 @@ class LambdaPostHook : public torch::autograd::FunctionPostHook {
return fn_(outputs, inputs);
}
void compiled_args(CompiledNodeArgs& args) override {}
void compiled_args(CompiledNodeArgs& args) const override {}
protected:
std::function<variable_list(const variable_list&, const variable_list&)> fn_;