Skip to content

[autoparallel] Patch meta information of torch.nn.functional.softmax and torch.nn.Softmax#2674

Merged
YuliangLiu0306 merged 59 commits intohpcaitech:mainfrom
Cypher30:feature/softmax_metainfo
Feb 13, 2023
Merged

[autoparallel] Patch meta information of torch.nn.functional.softmax and torch.nn.Softmax#2674
YuliangLiu0306 merged 59 commits intohpcaitech:mainfrom
Cypher30:feature/softmax_metainfo

Conversation

@Cypher30
Copy link
Copy Markdown
Contributor

@Cypher30 Cypher30 commented Feb 13, 2023

📌 Checklist before creating the PR

  • I have created an issue for this PR for traceability
  • The title follows the standard format: [doc/gemini/tensor/...]: A concise description
  • I have added relevant tags if possible for us to better distinguish different PRs

🚨 Issue number

Link this PR to your issue with words like fixed to automatically close the linked issue upon merge

e.g. fixed #1234, closed #1234, resolved #1234
Resolved #2629

📝 What does this PR do?

Summarize your work here.
if you have any plots/diagrams/screenshots/tables, please attach them here.

In this PR, I patch the meta information of torch.nn.functional.softmax and torch.nn.Softmax

💥 Checklist before requesting a review

  • I have linked my PR to an issue (instruction)
  • My issue clearly describes the problem/feature/proposal, with diagrams/charts/table/code if possible
  • I have performed a self-review of my code
  • I have added thorough tests.
  • I have added docstrings for all the functions/methods I implemented

⭐️ Do you enjoy contributing to Colossal-AI?

  • 🌝 Yes, I do.
  • 🌚 No, I don't.

Tell us more if you don't enjoy contributing to Colossal-AI.

Cypher30 and others added 30 commits July 14, 2022 16:07
@Cypher30 Cypher30 added Run Build and Test auto-parallel related to the auto-parallel feature labels Feb 13, 2023
@github-actions
Copy link
Copy Markdown
Contributor

The code coverage for the changed files is 34%.

Click me to view the complete report
Name                                                                                   Stmts   Miss  Cover
----------------------------------------------------------------------------------------------------------
colossalai/auto_parallel/meta_profiler/meta_registry/activation.py                        42     30    29%
tests/test_auto_parallel/test_tensor_shard/test_metainfo/test_activation_metainfo.py      62     39    37%
----------------------------------------------------------------------------------------------------------
TOTAL                                                                                    104     69    34%

@YuliangLiu0306 YuliangLiu0306 merged commit 40c916b into hpcaitech:main Feb 13, 2023
@github-actions
Copy link
Copy Markdown
Contributor

The code coverage for the changed files is 34%.

Click me to view the complete report
Name                                                                                   Stmts   Miss  Cover
----------------------------------------------------------------------------------------------------------
colossalai/auto_parallel/meta_profiler/meta_registry/activation.py                        42     30    29%
tests/test_auto_parallel/test_tensor_shard/test_metainfo/test_activation_metainfo.py      62     39    37%
----------------------------------------------------------------------------------------------------------
TOTAL                                                                                    104     69    34%

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

auto-parallel related to the auto-parallel feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[FEATURE]: Patch meta information of torch.nn.functional.softmax()

2 participants