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

[CI] Merge Matmul class with its Benchmark class #1086

Merged
merged 2 commits into from
Feb 7, 2025

Conversation

yzhang93
Copy link
Contributor

@yzhang93 yzhang93 commented Feb 6, 2025

Currently, there are Matmul/MatmulTranspseB/MatmulTransposeA classes and their Benchmark classes separately. A lot of codes are duplicated between them. This also brings extra efforts and more codes if we introduce other classes, e.g., mmt4d.

This PR removes the Benchmark classes while adding an argument "run_benchmark" to the existing Matmul/MatmulTranspseB/MatmulTransposeA classes.

@@ -2183,13 +2052,13 @@ def __init__(self):

if (transpose_a, transpose_b) == (False, False):
NumericTestClass = Matmul
Copy link
Contributor

@Yu-Zhewen Yu-Zhewen Feb 6, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe just use TestClass here, as NumericTestClass and BenchmarkTestClass are now pointing to the same thing.

@yzhang93 yzhang93 force-pushed the refactor_performance_test branch from a881253 to ce178f0 Compare February 6, 2025 23:47
@yzhang93 yzhang93 merged commit 416a925 into nod-ai:main Feb 7, 2025
7 checks passed
@yzhang93 yzhang93 deleted the refactor_performance_test branch February 11, 2025 02:02
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