[Dev] Move bitblas package to the project root (#68)

* chore: Update support matrix in README * Move bitblas package to root * Remove unused code files
microsoft · Jul 3, 2024 · 33869d4 · 33869d4
1 parent 74d8f3c
commit 33869d4
Show file tree

Hide file tree

Showing 82 changed files with 27 additions and 115 deletions.
diff --git a/python/bitblas/__init__.py → bitblas/__init__.py b/python/bitblas/__init__.py → bitblas/__init__.py
@@ -11,7 +11,7 @@
     sys.path.insert(0, install_tvm_path)
 
 develop_tvm_path = os.path.join(
-    os.path.dirname(os.path.abspath(__file__)), "..", "..", "3rdparty", "tvm", "python")
+    os.path.dirname(os.path.abspath(__file__)), "..", "3rdparty", "tvm", "python")
 if os.path.exists(develop_tvm_path) and develop_tvm_path not in sys.path:
     os.environ["PYTHONPATH"] = develop_tvm_path + ":" + os.environ.get("PYTHONPATH", "")
     sys.path.insert(0, develop_tvm_path)

diff --git a/python/bitblas/base/__init__.py → bitblas/base/__init__.py b/python/bitblas/base/__init__.py → bitblas/base/__init__.py
diff --git a/python/bitblas/base/analysis.py → bitblas/base/analysis.py b/python/bitblas/base/analysis.py → bitblas/base/analysis.py
diff --git a/python/bitblas/base/common_schedules.py → bitblas/base/common_schedules.py b/python/bitblas/base/common_schedules.py → bitblas/base/common_schedules.py
diff --git a/python/bitblas/base/roller/__init__.py → bitblas/base/roller/__init__.py b/python/bitblas/base/roller/__init__.py → bitblas/base/roller/__init__.py
diff --git a/python/bitblas/base/roller/arch/__init__.py → bitblas/base/roller/arch/__init__.py b/python/bitblas/base/roller/arch/__init__.py → bitblas/base/roller/arch/__init__.py
diff --git a/python/bitblas/base/roller/arch/arch_base.py → bitblas/base/roller/arch/arch_base.py b/python/bitblas/base/roller/arch/arch_base.py → bitblas/base/roller/arch/arch_base.py
diff --git a/python/bitblas/base/roller/arch/cpu.py → bitblas/base/roller/arch/cpu.py b/python/bitblas/base/roller/arch/cpu.py → bitblas/base/roller/arch/cpu.py
diff --git a/python/bitblas/base/roller/arch/cuda.py → bitblas/base/roller/arch/cuda.py b/python/bitblas/base/roller/arch/cuda.py → bitblas/base/roller/arch/cuda.py
diff --git a/python/bitblas/base/roller/bestfit.py → bitblas/base/roller/bestfit.py b/python/bitblas/base/roller/bestfit.py → bitblas/base/roller/bestfit.py
diff --git a/python/bitblas/base/roller/hint.py → bitblas/base/roller/hint.py b/python/bitblas/base/roller/hint.py → bitblas/base/roller/hint.py
diff --git a/python/bitblas/base/roller/node.py → bitblas/base/roller/node.py b/python/bitblas/base/roller/node.py → bitblas/base/roller/node.py
diff --git a/...on/bitblas/base/roller/policy/__init__.py → bitblas/base/roller/policy/__init__.py b/...on/bitblas/base/roller/policy/__init__.py → bitblas/base/roller/policy/__init__.py
diff --git a/python/bitblas/base/roller/policy/common.py → bitblas/base/roller/policy/common.py b/python/bitblas/base/roller/policy/common.py → bitblas/base/roller/policy/common.py
diff --git a/python/bitblas/base/roller/policy/default.py → bitblas/base/roller/policy/default.py b/python/bitblas/base/roller/policy/default.py → bitblas/base/roller/policy/default.py
diff --git a/.../bitblas/base/roller/policy/tensorcore.py → bitblas/base/roller/policy/tensorcore.py b/.../bitblas/base/roller/policy/tensorcore.py → bitblas/base/roller/policy/tensorcore.py
diff --git a/python/bitblas/base/roller/rasterization.py → bitblas/base/roller/rasterization.py b/python/bitblas/base/roller/rasterization.py → bitblas/base/roller/rasterization.py
diff --git a/...s/base/roller/shape_inference/__init__.py → ...s/base/roller/shape_inference/__init__.py b/...s/base/roller/shape_inference/__init__.py → ...s/base/roller/shape_inference/__init__.py
diff --git a/...las/base/roller/shape_inference/common.py → ...las/base/roller/shape_inference/common.py b/...las/base/roller/shape_inference/common.py → ...las/base/roller/shape_inference/common.py
diff --git a/...itblas/base/roller/shape_inference/tir.py → bitblas/base/roller/shape_inference/tir.py b/...itblas/base/roller/shape_inference/tir.py → bitblas/base/roller/shape_inference/tir.py
diff --git a/python/bitblas/base/schedule_rule.py → bitblas/base/schedule_rule.py b/python/bitblas/base/schedule_rule.py → bitblas/base/schedule_rule.py
diff --git a/python/bitblas/base/transform.py → bitblas/base/transform.py b/python/bitblas/base/transform.py → bitblas/base/transform.py
diff --git a/python/bitblas/base/utils.py → bitblas/base/utils.py b/python/bitblas/base/utils.py → bitblas/base/utils.py
diff --git a/python/bitblas/cache/__init__.py → bitblas/cache/__init__.py b/python/bitblas/cache/__init__.py → bitblas/cache/__init__.py
diff --git a/python/bitblas/cache/operator.py → bitblas/cache/operator.py b/python/bitblas/cache/operator.py → bitblas/cache/operator.py
diff --git a/python/bitblas/generator.py → bitblas/generator.py b/python/bitblas/generator.py → bitblas/generator.py
diff --git a/python/bitblas/gpu/__init__.py → bitblas/gpu/__init__.py b/python/bitblas/gpu/__init__.py → bitblas/gpu/__init__.py
diff --git a/python/bitblas/gpu/base.py → bitblas/gpu/base.py b/python/bitblas/gpu/base.py → bitblas/gpu/base.py
diff --git a/python/bitblas/gpu/element_wise.py → bitblas/gpu/element_wise.py b/python/bitblas/gpu/element_wise.py → bitblas/gpu/element_wise.py
diff --git a/python/bitblas/gpu/fallback.py → bitblas/gpu/fallback.py b/python/bitblas/gpu/fallback.py → bitblas/gpu/fallback.py
diff --git a/python/bitblas/gpu/gemv.py → bitblas/gpu/gemv.py b/python/bitblas/gpu/gemv.py → bitblas/gpu/gemv.py
diff --git a/python/bitblas/gpu/gemv_dequantize.py → bitblas/gpu/gemv_dequantize.py b/python/bitblas/gpu/gemv_dequantize.py → bitblas/gpu/gemv_dequantize.py
diff --git a/python/bitblas/gpu/general_reduction.py → bitblas/gpu/general_reduction.py b/python/bitblas/gpu/general_reduction.py → bitblas/gpu/general_reduction.py
diff --git a/python/bitblas/gpu/intrin/__init__.py → bitblas/gpu/intrin/__init__.py b/python/bitblas/gpu/intrin/__init__.py → bitblas/gpu/intrin/__init__.py
diff --git a/python/bitblas/gpu/intrin/lop3.py → bitblas/gpu/intrin/lop3.py b/python/bitblas/gpu/intrin/lop3.py → bitblas/gpu/intrin/lop3.py
diff --git a/python/bitblas/gpu/matmul.py → bitblas/gpu/matmul.py b/python/bitblas/gpu/matmul.py → bitblas/gpu/matmul.py
diff --git a/python/bitblas/gpu/matmul_analysis.py → bitblas/gpu/matmul_analysis.py b/python/bitblas/gpu/matmul_analysis.py → bitblas/gpu/matmul_analysis.py
diff --git a/python/bitblas/gpu/matmul_mma.py → bitblas/gpu/matmul_mma.py b/python/bitblas/gpu/matmul_mma.py → bitblas/gpu/matmul_mma.py
diff --git a/python/bitblas/gpu/matmul_mma_dequantize.py → bitblas/gpu/matmul_mma_dequantize.py b/python/bitblas/gpu/matmul_mma_dequantize.py → bitblas/gpu/matmul_mma_dequantize.py
diff --git a/python/bitblas/gpu/matmul_wmma.py → bitblas/gpu/matmul_wmma.py b/python/bitblas/gpu/matmul_wmma.py → bitblas/gpu/matmul_wmma.py
diff --git a/python/bitblas/gpu/reduction.py → bitblas/gpu/reduction.py b/python/bitblas/gpu/reduction.py → bitblas/gpu/reduction.py
diff --git a/python/bitblas/gpu/rmsnorm.py → bitblas/gpu/rmsnorm.py b/python/bitblas/gpu/rmsnorm.py → bitblas/gpu/rmsnorm.py
diff --git a/python/bitblas/gpu/transpose.py → bitblas/gpu/transpose.py b/python/bitblas/gpu/transpose.py → bitblas/gpu/transpose.py
diff --git a/python/bitblas/gpu/utils.py → bitblas/gpu/utils.py b/python/bitblas/gpu/utils.py → bitblas/gpu/utils.py
diff --git a/python/bitblas/module/__init__.py → bitblas/module/__init__.py b/python/bitblas/module/__init__.py → bitblas/module/__init__.py
diff --git a/python/bitblas/ops/__init__.py → bitblas/ops/__init__.py b/python/bitblas/ops/__init__.py → bitblas/ops/__init__.py
diff --git a/python/bitblas/ops/general_matmul.py → bitblas/ops/general_matmul.py b/python/bitblas/ops/general_matmul.py → bitblas/ops/general_matmul.py
diff --git a/python/bitblas/ops/general_matmul_splitk.py → bitblas/ops/general_matmul_splitk.py b/python/bitblas/ops/general_matmul_splitk.py → bitblas/ops/general_matmul_splitk.py
diff --git a/python/bitblas/ops/impl/__init__.py → bitblas/ops/impl/__init__.py b/python/bitblas/ops/impl/__init__.py → bitblas/ops/impl/__init__.py
diff --git a/.../ops/impl/batch_matmul_dequantize_impl.py → .../ops/impl/batch_matmul_dequantize_impl.py b/.../ops/impl/batch_matmul_dequantize_impl.py → .../ops/impl/batch_matmul_dequantize_impl.py
diff --git a/python/bitblas/ops/impl/batch_matmul_impl.py → bitblas/ops/impl/batch_matmul_impl.py b/python/bitblas/ops/impl/batch_matmul_impl.py → bitblas/ops/impl/batch_matmul_impl.py
diff --git a/...on/bitblas/ops/impl/convolution2d_impl.py → bitblas/ops/impl/convolution2d_impl.py b/...on/bitblas/ops/impl/convolution2d_impl.py → bitblas/ops/impl/convolution2d_impl.py
diff --git a/...bitblas/ops/impl/ladder_permutate_impl.py → bitblas/ops/impl/ladder_permutate_impl.py b/...bitblas/ops/impl/ladder_permutate_impl.py → bitblas/ops/impl/ladder_permutate_impl.py
diff --git a/...n/bitblas/ops/impl/lop3_permutate_impl.py → bitblas/ops/impl/lop3_permutate_impl.py b/...n/bitblas/ops/impl/lop3_permutate_impl.py → bitblas/ops/impl/lop3_permutate_impl.py
diff --git a/...itblas/ops/impl/matmul_dequantize_impl.py → bitblas/ops/impl/matmul_dequantize_impl.py b/...itblas/ops/impl/matmul_dequantize_impl.py → bitblas/ops/impl/matmul_dequantize_impl.py
diff --git a/...ops/impl/matmul_dequantize_splitk_impl.py → ...ops/impl/matmul_dequantize_splitk_impl.py b/...ops/impl/matmul_dequantize_splitk_impl.py → ...ops/impl/matmul_dequantize_splitk_impl.py
diff --git a/python/bitblas/ops/impl/matmul_impl.py → bitblas/ops/impl/matmul_impl.py b/python/bitblas/ops/impl/matmul_impl.py → bitblas/ops/impl/matmul_impl.py
diff --git a/...on/bitblas/ops/impl/matmul_splitk_impl.py → bitblas/ops/impl/matmul_splitk_impl.py b/...on/bitblas/ops/impl/matmul_splitk_impl.py → bitblas/ops/impl/matmul_splitk_impl.py
diff --git a/.../bitblas/ops/impl/param_permutate_impl.py → bitblas/ops/impl/param_permutate_impl.py b/.../bitblas/ops/impl/param_permutate_impl.py → bitblas/ops/impl/param_permutate_impl.py
diff --git a/python/bitblas/ops/ladder_permutate.py → bitblas/ops/ladder_permutate.py b/python/bitblas/ops/ladder_permutate.py → bitblas/ops/ladder_permutate.py
diff --git a/python/bitblas/ops/lop3_permutate.py → bitblas/ops/lop3_permutate.py b/python/bitblas/ops/lop3_permutate.py → bitblas/ops/lop3_permutate.py
diff --git a/python/bitblas/ops/matmul.py → bitblas/ops/matmul.py b/python/bitblas/ops/matmul.py → bitblas/ops/matmul.py
diff --git a/python/bitblas/ops/matmul_dequantize.py → bitblas/ops/matmul_dequantize.py b/python/bitblas/ops/matmul_dequantize.py → bitblas/ops/matmul_dequantize.py
diff --git a/python/bitblas/ops/operator.py → bitblas/ops/operator.py b/python/bitblas/ops/operator.py → bitblas/ops/operator.py
diff --git a/python/bitblas/ops/param_permutate.py → bitblas/ops/param_permutate.py b/python/bitblas/ops/param_permutate.py → bitblas/ops/param_permutate.py
diff --git a/python/bitblas/quantization/__init__.py → bitblas/quantization/__init__.py b/python/bitblas/quantization/__init__.py → bitblas/quantization/__init__.py
diff --git a/python/bitblas/quantization/quantization.py → bitblas/quantization/quantization.py b/python/bitblas/quantization/quantization.py → bitblas/quantization/quantization.py
diff --git a/python/bitblas/quantization/utils.py → bitblas/quantization/utils.py b/python/bitblas/quantization/utils.py → bitblas/quantization/utils.py
diff --git a/python/bitblas/relax/op/interleave_weight.py → bitblas/relax/op/interleave_weight.py b/python/bitblas/relax/op/interleave_weight.py → bitblas/relax/op/interleave_weight.py
diff --git a/python/bitblas/relax/transform/__init__.py → bitblas/relax/transform/__init__.py b/python/bitblas/relax/transform/__init__.py → bitblas/relax/transform/__init__.py
diff --git a/.../relax/transform/annotate_decode_block.py → .../relax/transform/annotate_decode_block.py b/.../relax/transform/annotate_decode_block.py → .../relax/transform/annotate_decode_block.py
diff --git a/.../relax/transform/weight_only_propagate.py → .../relax/transform/weight_only_propagate.py b/.../relax/transform/weight_only_propagate.py → .../relax/transform/weight_only_propagate.py
diff --git a/python/bitblas/testing/__init__.py → bitblas/testing/__init__.py b/python/bitblas/testing/__init__.py → bitblas/testing/__init__.py
diff --git a/python/bitblas/utils/__init__.py → bitblas/utils/__init__.py b/python/bitblas/utils/__init__.py → bitblas/utils/__init__.py
diff --git a/python/bitblas/utils/post_process.py → bitblas/utils/post_process.py b/python/bitblas/utils/post_process.py → bitblas/utils/post_process.py
diff --git a/python/bitblas/utils/target_detector.py → bitblas/utils/target_detector.py b/python/bitblas/utils/target_detector.py → bitblas/utils/target_detector.py
diff --git a/python/bitblas/utils/tensor_adapter.py → bitblas/utils/tensor_adapter.py b/python/bitblas/utils/tensor_adapter.py → bitblas/utils/tensor_adapter.py
diff --git a/python/bitblas/wrapper/__init__.py → bitblas/wrapper/__init__.py b/python/bitblas/wrapper/__init__.py → bitblas/wrapper/__init__.py
diff --git a/python/bitblas/wrapper/general.py → bitblas/wrapper/general.py b/python/bitblas/wrapper/general.py → bitblas/wrapper/general.py
diff --git a/install.sh b/install.sh
@@ -0,0 +1,26 @@
+#!/bin/bash
+
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+# install requirements
+pip install -r requirements.txt
+
+# install llvm
+apt-get install llvm-10
+
+# clone and build tvm
+git submodule update --init --recursive
+
+cd 3rdparty/tvm
+mkdir build
+cp cmake/config.cmake build
+cd build
+echo "set(USE_LLVM llvm-config-10)" >> config.cmake && echo "set(USE_CUDA ON)" >> config.cmake
+
+cmake .. && make -j && cd ../../..
+
+echo "export TVM_HOME=$(pwd)/3rdparty/tvm" >> ~/.bashrc
+echo "export PYTHONPATH=\$TVM_HOME/python:$(pwd)/python:\$PYTHONPATH" >> ~/.bashrc
+
+source ~/.bashrc
diff --git a/python/bitblas_cli.py b/python/bitblas_cli.py
diff --git a/testing/python/dsl/test_auto_normalized_tensorcore.py b/testing/python/dsl/test_auto_normalized_tensorcore.py