[Dev] Move Relax Pass from testing to integration #77
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request primarily involves refactoring and code quality improvements in the
bitblas/ops/impl
package. The changes include the introduction of abstract base classes for TIR script emitters and selectors, refactoring of thematmul_nt
function into a class with methods for better organization and readability, and minor changes to suppress linting warnings.Refactoring and code quality improvements:
bitblas/ops/impl/__init__.py
: Suppressed a linting warning by adding# noqa: F401
to the import statement.bitblas/ops/impl/base.py
: Introduced abstract base classesTIRScriptEmitter
andTIRScriptSelector
for TIR script emitters and selectors respectively. These classes enforce the implementation ofemit
andselect
methods in any derived classes.bitblas/ops/impl/batch_matmul_impl.py
: Refactored thematmul_nt
function into theBatchMatMulEmitter
class with multiple methods for better organization and readability. This class inherits from theTIRScriptEmitter
base class. A similar refactoring was done for thematmul
function, which was transformed into theBatchMatMulSelector
class, inheriting from theTIRScriptSelector
base class. Theselect_implementation
function was also simplified to use the newly createdBatchMatMulSelector
class. [1] [2]