• Lei Wang's avatar
    [Language] Introduce `T.alloc_var` to define a variable like `int var;` (#255) · c770a58f
    Lei Wang authored
    * [Enhancement] Add matrix multiplication functions for integer and float variables in Cython JIT
    
    - Introduced `matmul_int_variable` and `matmul_float_variable` functions to support matrix multiplication with dynamic shapes and additional parameters.
    - Implemented corresponding `run_matmul_int_variable` and `run_matmul_float_variable` functions for testing.
    - Updated test cases to validate the new matrix multiplication implementations.
    - Enhanced error handling in library initialization and compilation processes across various modules.
    - Improved dynamic memory handling in CUDA kernel initialization to provide better error reporting.
    
    * lint fix
    
    * optimize
    
    * Support var defiine
    
    * lint fix
    
    * Update TVM submodule and add alloc_variable function to allocate local variables in TileLang
    
    - Updated the TVM submodule to the latest commit.
    - Introduced `alloc_variable` function in `allocate.py` to support local variable allocation with specified data types and scopes.
    
    * lint fix
    
    * Refactor variable allocation functions for consistency
    
    - Renamed `alloc_variable` to `alloc_var` across multiple files for improved consistency.
    - Updated corresponding test functions to reflect the new naming convention.
    - Adjusted imports in `__init__.py` to align with the changes.
    c770a58f
codegen_cuda.cc 60.8 KB