"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a1f36ee3ef4ae1bf98bd260e539197259aa981c1"
  • Lei Wang's avatar
    [Refactor] Refactor `jit` to `_JitImplementation` to support `@tilelang.jit` (#502) · 8c8d8ca2
    Lei Wang authored
    * [Refactor] Rename `jit` class to `_JitImplementation` and improve debug path handling
    
    * Refactored the `jit` class to `_JitImplementation` for clarity and encapsulation.
    * Enhanced handling of `debug_root_path` to ensure it is correctly set as an absolute path when provided.
    * Updated the public `jit` function to serve as a decorator interface, allowing for both default and configured usage.
    * Added validation to ensure input tensors are contiguous in the Cython wrapper, improving error handling.
    
    * [Refactor] Improve formatting and handling in `_JitImplementation` and `jit` function
    
    * Refactored the `_JitImplementation` class to enhance readability by adjusting comment formatting and consolidating conditions for setting `debug_root_path`.
    * Updated the `jit` function signature for better alignment and clarity in parameter definitions.
    * Ensured consistent spacing and comments throughout the code for improved maintainability.
    
    * [Refactor] Update GEMM test parameters for performance optimization
    
    * Set num_stages to 0 and adjusted matrix dimensions in the GEMM test function to enhance performance and consistency across tests in test_tilelang_jit_gemm.py.
    * Reduced the number of threads used in the test to align with the updated configuration, improving overall test efficiency.
    
    * [Refactor] Enhance buffer error logging in layout inference
    
    * Updated the warning message in layout inference to provide clearer context when a buffer cannot be inferred due to its absence in the use list. This change improves the clarity of error reporting during layout inference operations.
    * Refactored tensor handling in the Cython wrapper to ensure input tensors are checked for contiguity before processing, enhancing error handling and robustness in tensor management.
    
    * bugfix
    8c8d8ca2
test_tilelang_jit_gemm.py 2.96 KB