Skip to content

feat: add Module::children() to expose direct submodules#1162

Merged
wooway777 merged 1 commit into
InfiniTensor:mainfrom
qinyiqun:linear_refactor
May 12, 2026
Merged

feat: add Module::children() to expose direct submodules#1162
wooway777 merged 1 commit into
InfiniTensor:mainfrom
qinyiqun:linear_refactor

Conversation

@qinyiqun
Copy link
Copy Markdown
Collaborator

Add a public accessor for the direct submodule map so that recursive traversal (e.g. process_weights_after_loading) can iterate only immediate children instead of the fully flattened modules_dict().

Add a public accessor for the direct submodule map so that recursive
traversal (e.g. process_weights_after_loading) can iterate only
immediate children instead of the fully flattened modules_dict().
@qinyiqun qinyiqun requested a review from a team May 12, 2026 02:14
@wooway777 wooway777 merged commit 90fd438 into InfiniTensor:main May 12, 2026
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants