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

[Dev] Refactor Backend Dispatch and Kernel Wrap Related Design #83

Merged
merged 33 commits into from
Jul 11, 2024

Conversation

LeiWang1999
Copy link
Contributor

@LeiWang1999 LeiWang1999 commented Jul 10, 2024

This pull request mainly includes changes to improve the architecture of the project, modify the CI workflow, and update the third-party notices. The most significant changes are the addition of the LibraryGenerator class, the modification of the BaseWrapper class, and the changes in the import statements to reflect the new project structure.

Improvements to project architecture:

  • bitblas/base/roller/__init__.py, bitblas/base/roller/policy/default.py, bitblas/base/roller/policy/tensorcore.py, bitblas/base/utils.py: Modified the import statements to reflect the new project structure. [1] [2] [3] [4]
  • bitblas/builder/lib_generator/__init__.py: Added the LibraryGenerator class, which compiles and loads libraries, and handles the source and library paths.
  • bitblas/builder/wrapper/__init__.py, bitblas/builder/wrapper/base.py: Modified the BaseWrapper class to include the wrap method. [1] [2]
  • bitblas/cache/operator.py: Updated the method signatures and the method calls to use srcpath and libpath instead of src_name and lib_name. [1] [2] [3]
  • bitblas/generator.py: Removed the BitBLASGenerator class.
  • bitblas/ops/general_matmul/__init__.py: Renamed from bitblas/ops/general_matmul.py and updated the import statements to reflect the new project structure. Also modified the MatmulConfig class to inherit from OperatorConfig. [1] [2] [3]

CI workflow modification:

Third-party notices update:

Ref to dicussion #81

@LeiWang1999 LeiWang1999 merged commit 66304d2 into microsoft:main Jul 11, 2024
6 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.

1 participant