mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-25 07:14:16 +08:00
Lei Wang
fc4bd452b3
* [Layout] Add IsCompletedReplicated method and enhance layout inference in ParallelOpNode - Introduced IsCompletedReplicated method in FragmentNode to check if a buffer is fully replicated. - Enhanced InferLayout in ParallelOpNode to handle layout inference for replicated buffers, ensuring only fragment[0] access is allowed. - Updated error handling for non-zero index access in fragment buffers to improve robustness. * [Layout] Improve code formatting and readability in layout.cc and parallel.cc - Enhanced formatting in FragmentNode's IsCompletedReplicated method for better clarity. - Updated InferLayout method in ParallelOpNode to improve code readability by adjusting line breaks and indentation. - Ensured consistent formatting across conditional statements and comments for improved maintainability. * updt * optimize const index related op * bug fix * reduce gdn test * test fix * lintfix * lint fix * test fix
27 lines
550 B
Python
27 lines
550 B
Python
import tilelang.testing
|
|
import example_gemm_autotune
|
|
import example_gemm_intrinsics
|
|
import example_gemm_schedule
|
|
import example_gemm
|
|
|
|
|
|
def test_example_gemm_autotune():
|
|
# enable roller for fast tuning
|
|
example_gemm_autotune.main(M=1024, N=1024, K=1024, with_roller=True)
|
|
|
|
|
|
def test_example_gemm_intrinsics():
|
|
example_gemm_intrinsics.main(M=1024, N=1024, K=1024)
|
|
|
|
|
|
def test_example_gemm_schedule():
|
|
example_gemm_schedule.main()
|
|
|
|
|
|
def test_example_gemm():
|
|
example_gemm.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|