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 testing scripts and fix security issues #72

Merged
merged 20 commits into from
Jul 4, 2024

Conversation

LeiWang1999
Copy link
Contributor

This pull request contains changes primarily aimed at modifying the import statements of the tvm module in various files and adjusting the path settings for the tvm module in the bitblas/__init__.py file. Additionally, there are changes to the bitblas/ops/general_matmul.py and bitblas/module/__init__.py files to modify the function logic.

Changes to the import statements:

  • In multiple files, the import statement for the tvm module has been changed from import tvm to from bitblas import tvm. This change has been made in the following files: bitblas/base/roller/arch/cpu.py [1] bitblas/base/roller/arch/cuda.py [2] bitblas/base/roller/node.py [3] bitblas/base/roller/policy/default.py [4] bitblas/base/roller/policy/tensorcore.py [5] bitblas/base/transform.py [6] bitblas/base/utils.py [7] bitblas/cache/operator.py [8] bitblas/gpu/intrin/lop3.py [9] bitblas/gpu/rmsnorm.py [10] bitblas/ops/general_matmul.py [11] bitblas/ops/impl/batch_matmul_dequantize_impl.py [12] bitblas/ops/impl/batch_matmul_impl.py [13] bitblas/ops/impl/convolution2d_impl.py [14] bitblas/ops/impl/matmul_dequantize_impl.py [15] bitblas/ops/impl/matmul_dequantize_splitk_impl.py [16] bitblas/ops/impl/matmul_impl.py [17].

Changes to the path settings:

  • In the bitblas/__init__.py file, the path settings for the tvm module have been adjusted. The path is now appended with /python before being added to the PYTHONPATH environment variable and the Python system path.

Changes to function logic:

  • In bitblas/ops/general_matmul.py, the logic for the __legalize_propagate function has been adjusted to handle cases where propagate_b is not transformed.
  • In bitblas/module/__init__.py, the forward function has been modified to adjust the order of operations and the arguments passed to the lib.call method.

ref security issue: #71 #70

@LeiWang1999 LeiWang1999 marked this pull request as ready for review July 4, 2024 09:24
@LeiWang1999 LeiWang1999 merged commit 8804d77 into microsoft:main Jul 4, 2024
3 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