• Lei Wang's avatar
    [Refactor] Add kernel selection option for GEMM v1 in environment settings (#1200) · 8fbe1b3a
    Lei Wang authored
    * Add kernel selection option for GEMM v1 in environment settings
    
    - Introduced `TILELANG_USE_GEMM_V1` environment variable to control the selection of GEMM version.
    - Added `use_gemm_v1` method in the `Environment` class to determine if GEMM v1 should be used based on the environment variable.
    - Updated GEMM function assignment to default to v2, allowing for v1 to be forced via the new environment variable.
    
    * bug fix
    
    * Add kernel selection option for GEMM in environment settings
    
    - Introduced `TILELANG_USE_GEMM_V1` environment variable to allow users to select between GEMM v1 and v2 implementations.
    - Updated `gemm` function to default to v2 but switch to v1 if the environment variable is set to a truthy value.
    - Added a method `use_gemm_v1` in the `Environment` class to facilitate this selection based on the environment variable.
    
    * Refactor GEMM macro generator to use BufferRegion instead of Buffer
    
    - Updated `wgmma` a...
    8fbe1b3a
test_tilelang_dynamic_symbolic.py 15.2 KB