Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[tripy] Sequential module feature branch #321

Merged
merged 29 commits into from
Nov 12, 2024
Merged

[tripy] Sequential module feature branch #321

merged 29 commits into from
Nov 12, 2024

Conversation

farazkh80
Copy link
Collaborator

@farazkh80 farazkh80 commented Oct 29, 2024

Sequential similar to torch.nn.sequential for addressing #295

Things done

  • new frontend module called tp.Sequential
  • Unit tests under tests/frontend/module/test_sequential.py
  • integration tests and comparison with torch under

Supports nested tp.Sequential and most list operations other than modifications. tp.Sequential can not be modified after creation.

Faraz Khoubsirat added 5 commits October 29, 2024 21:36
Signed-off-by: Faraz Khoubsirat <[email protected]>
Signed-off-by: Faraz Khoubsirat <[email protected]>
Signed-off-by: Faraz Khoubsirat <[email protected]>
Signed-off-by: Faraz Khoubsirat <[email protected]>
farazkh80 and others added 3 commits November 4, 2024 18:18
Co-authored-by: pranavm-nvidia <[email protected]>
Signed-off-by: Faraz <[email protected]>
Co-authored-by: pranavm-nvidia <[email protected]>
Signed-off-by: Faraz <[email protected]>
Co-authored-by: pranavm-nvidia <[email protected]>
Signed-off-by: Faraz <[email protected]>
@farazkh80
Copy link
Collaborator Author

Addressed PR changes, and added some more tests for Dict mode of Sequential @pranavm-nvidia

@farazkh80
Copy link
Collaborator Author

Addressed final MR change requests and fixed tests @pranavm-nvidia

@farazkh80 farazkh80 merged commit bbef6e1 into main Nov 12, 2024
1 check passed
@farazkh80 farazkh80 deleted the tripy-sequential branch November 12, 2024 21:33
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