"docs/vscode:/vscode.git/clone" did not exist on "21e3f27e8eb9fe0b6b2afa1f08622f10dff30ece"
  • Lei Wang's avatar
    [Bugfix] Add dynamic shape support with out_idx in Cython JIT kernel compilation (#185) · d34601ab
    Lei Wang authored
    * Optimize CMake build process with dynamic job count calculation
    
    - Modify build_csrc function to use 90% of available CPU cores
    - Ensure at least one job is used during compilation
    - Improve build performance by dynamically adjusting parallel job count
    
    * Optimize build_csrc function with multiprocessing module
    
    - Replace os.cpu_count() with multiprocessing.cpu_count()
    - Maintain existing 90% CPU utilization logic
    - Improve CPU core count calculation for build process
    
    * Add dynamic shape support with out_idx in Cython JIT kernel compilation
    
    - Implement `run_cython_dynamic_shape_with_out_idx` function in test_tilelang_jit_gemm_cython.py
    - Update Cython wrapper to handle dynamic symbolic shapes during tensor allocation
    - Add support for resolving dynamic shape dimensions using input tensor references
    - Enhance flexibility of JIT kernel compilation with symbolic shape handling
    
    * Enhance error reporting for dynamic symbolic shape resolution in Cython JIT kernel
    
    - Add detailed error message when a dynamic symbolic dimension is not found in dynamic_symbolic_map
    - Improve debugging by providing context about missing symbolic dimensions
    - Maintain existing dynamic shape resolution logic
    d34601ab
test_tilelang_jit_gemm_cython.py 12.6 KB