Skip to content

Conversation

@chinganc
Copy link
Collaborator

@chinganc chinganc commented Jan 15, 2025

This update fixes multiple bugs.

  1. Sometimes TraceGraph does not follow heap ordering (previous commit 6c4c52d)

  2. When @Bundle(trainable=True) is applied to a class method, different class instances will unintentionally share the same parameter.

  3. When calling @Bundle class method(s) multiple times of a class instance, duplicated nodes of the class instance are created, which all point to the same class instance.

  4. When accessing the bundled method of a class instance, it returns a partial function instead of FunModule, which makes accessing properties of the bundled method difficult.

Fix multiple bugs

1. Sometimes TraceGraph does not follow heap ordering.
2. When @Bundle(trainable=True) is applied to a class method, different class instances will unintentionally share the same parameter.
3. When calling @Bundle class method(s) multiple times of a class instance, duplicated nodes of the class instance are created, which all point to the same class instance.
4. When accessing the bundled method of a class instance, it returns a partial function instead of FunModule, which makes accessing properties of the bundled method difficult.
@chinganc chinganc requested a review from allenanie January 15, 2025 00:51
@chinganc chinganc closed this Jan 15, 2025
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