Commit 8dc1d7df authored by alex_xiao's avatar alex_xiao Committed by LeiWang1999
Browse files

[Bugfix] Compile/"cached" still not loading cached kernel for example in example_mha_bwd (#339)

* [Dev] Add database mechanism to cache

* [Dev] Fix database cache and test for it

* [Dev] Refactor env.py to use TILELANG_CACHE_DIR and remove extra comment.

* [Refactor] Improve code formatting and readability in multiple files

* [Enhancement] Add execution backend options and improve kernel adapter initialization

* [Refactor] Rename cached function to cached_kernel and update related references

* [Enhancement] Enable target and target_host parameters in kernel loading and improve gemm test case

* [Enhancement] Update kernel compilation to specify execution backend as "cython"

* [Refactor] Rename cached_kernel to cached and update references in the codebase

* [Enhancement] Un-comment and add test cases for matrix multiplication correctness; improve kernel caching logic and remove redundant code

* [Refactor] Clean up code formatting and improve readability in cache and adapter modules

* [Refactor] Remove unused imports

* [Refactor] Update cached function signature to use PrimFunc and Optional types for improved type safety

* [Refactor] Update cached function calls to use PrimFunc and improve parameter handling

* [Refactor] Clean up import statements and improve code formatting in cache and kernel test files

* [Refactor] Update cache key generation to use function source code for hashing

* [Update] Update subproject commit for TVM

* [Update] Import inspect module in kernel_cache.py

* [Update] Change default execution backend to 'cython' in JITKernel

* redo tvm

* [Update] Add SHA256 hash for function parameters in KernelCache

* [Bugfix] fix merge error

* [Feat] Rearrange script for key generation

* [Bugfix] Delete extra files

* [Refactor] Improve code readability and formatting in kernel_cache.py

* [Refactor] Remove unused sorting function from KernelCache and simplify binary serialization

* Update submodule tvm
parent f005db9f
Subproject commit 3aca64b65b4c1586031da076ffbaa65280f21dac Subproject commit 2e033790663f0a470865b38da6931e9addb09238
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment