"test/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "7aac931eb67dd62eabc1f8fcd0fefabaff7c97a7"
  • Lei Wang's avatar
    [Feature] Add reduce_max corresponding tests (#236) · 12714155
    Lei Wang authored
    * [Feature] Add reduce_max functionality and corresponding tests
    
    * Introduced a new test file for the reduce_max operation in the tilelang language module.
    * Implemented the reduce_max functionality using T.prim_func, including local memory allocation and result copying.
    * Added tests for various input sizes and data types to ensure correctness of the reduce_max implementation.
    * Enhanced profiling assertions to validate the output against reference implementations.
    
    * Fix whitespace issues in reduce_max test file for improved readability
    12714155
test_tilelang_language_reduce_max.py 1.23 KB