saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix a PoliCheck finding in _hierarchical_ortmodule.py(#13462)

Browse source
This commit is contained in:
Vincent Wang 2022-10-27 06:45:18 +08:00 committed by GitHub
parent 490e4ddea5
commit 805ec459a0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/python/training/ortmodule/experimental/hierarchical_ortmodule/_hierarchical_ortmodule.py
View file

@ -112,7 +112,7 @@ class HierarchicalORTModule(torch.nn.Module):
# those nn.Module's with their recorded inputs.
# NOTE that if a module is not called from forward(), it will fail to be captured by this hook.
def recursive_hook(module):
# We cannot skip module in whitelist because it's possible that a module is called multiple times
# We cannot skip module in allowlist because it's possible that a module is called multiple times
# so that we still need to know the number of different input sets and use _IteratedORTModule to handle it.
handle_pool.append(module.register_forward_pre_hook(record_args))
for _, sub_module in module._modules.items():