[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
Showing
Please register or sign in to comment