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

Flops profiler support einops.einsum #6755

Merged
merged 13 commits into from
Dec 9, 2024

Conversation

lvhoaa
Copy link
Contributor

@lvhoaa lvhoaa commented Nov 17, 2024

  • Added support for FlopsProfiler to include einops.einsum operation
  • Added _patch_miscellaneous_operations() and _reload_miscellaneous_operations() to include this operation and potentially include other miscellaneous operations in the future
  • Added _einops_einsum_flops_compute() that mimic already-existed _einsum_flops_compute()

@lvhoaa lvhoaa requested a review from loadams as a code owner November 17, 2024 16:56
@lvhoaa
Copy link
Contributor Author

lvhoaa commented Nov 17, 2024

@microsoft-github-policy-service agree

bin/deepspeed Outdated Show resolved Hide resolved
@loadams
Copy link
Contributor

loadams commented Nov 26, 2024

Hi @lvhoaa - it looks like this PR (#6462) will be merged first that adds einops to the main requirements file, so you can remove the added to requirements-dev and that should fix the issues that we are seeing on this PR.

@loadams loadams enabled auto-merge December 9, 2024 17:40
@loadams loadams disabled auto-merge December 9, 2024 17:56
@loadams loadams merged commit 9a41cca into microsoft:master Dec 9, 2024
9 of 10 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