• alex_xiao's avatar
    [Feature] Add database storage for JITKernel cache with Cython and Ctypes adapters (#213) · e789808b
    alex_xiao authored
    
    
    * [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
    
    * Update tilelang/jit/kernel.py
    
    ---------
    Co-authored-by: default avatarLei Wang <34334180+LeiWang1999@users.noreply.github.com>
    e789808b
test_tilelang_kernel_mha_bwd.py 13.3 KB