-
Notifications
You must be signed in to change notification settings - Fork 24
Add GEMM & Expand #93
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
Conversation
Ziminli
commented
Oct 31, 2024
- Add GEMM operator
- Add Expand operator
- Add FP32 support for matmul operator
- Misc. changes to common utilities
operatorspy/tests/expand.py
Outdated
| ((3, 2, 4, 5), (3, 2, 1, 1), None, None), | ||
| ((32, 256, 112, 112), (32, 256, 112, 1), None, None), | ||
| # ((32, 256, 112, 112), (32, 1, 1, 1), None, None), | ||
| # ((32, 150, 51200), (32, 150, 1), None, None), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这俩测例是为什么要注释掉?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
因为GitHub CI 用CPU跑会比较慢,主要是用来测较大数据规模时性能的,如果不需要可以删掉
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
已删除