tilelang

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-22 06:50:25 +08:00

Files

T

Lei Wang 0f9bbd759c [Parallel][Infer] Free-mode chooses minimal replication between buffer-based and PlanLoopPartition (#1559 )

* [Enhancement] Improve layout inference in ParallelOpNode

* Enhanced the layout inference mechanism in ParallelOpNode to utilize two strategies: compute_loop_layout_from_buffer and PlanLoopPartition, selecting the one that minimizes replication while ensuring compatibility.
* Updated the logic to choose the best candidate layout based on replication size and containment checks.
* Refactored the HasKnownLayoutAnchor function to clarify its purpose in prioritizing buffer layouts.
* Added a new test case to validate the layout inference behavior, ensuring the correct fragments are generated in the output.

This update aims to optimize layout inference for parallel operations, improving performance and resource utilization.

* lint fix

* bug fix and refactor

* lint fix

2025-12-30 13:28:54 +08:00

test_tilelang_issue_96.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448 )

2025-12-17 14:41:25 +08:00

test_tilelang_issue_814.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448 )

2025-12-17 14:41:25 +08:00

test_tilelang_issue_830.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448 )

2025-12-17 14:41:25 +08:00

test_tilelang_issue_1001.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448 )