[Dev] Enhance TileLang Backend and fix a bug for INT4xINT2 #236
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 introduces a new utility function
retrieve_func_from_module
and refactors existing code to use this function for retrieving the main function from an IR module. The changes ensure a more consistent and reusable approach across multiple files.Key changes include:
Utility Function Addition:
bitblas/utils/__init__.py
: Addedretrieve_func_from_module
function to extract the main function from anIRModule
.Refactoring to Use New Utility Function:
bitblas/base/common_schedules.py
: Replaced direct access to the main function withretrieve_func_from_module
.bitblas/base/utils.py
: Updated multiple methods to useretrieve_func_from_module
for retrieving functions from modules.bitblas/ops/operator.py
: Replaced direct access to the main function withretrieve_func_from_module
in thehardware_aware_finetune
method.Code Clean-Up:
bitblas/ops/operator.py
: Removed theupdate_func
method as it is no longer needed due to the new utility function.