tilelang.contrib.ptodsl.sync¶
PTO synchronization compatibility helpers for generated TileLang kernels.
Functions¶
|
Lower TileLang's cross-core set flag for modes 0, 1, 2, and 4. |
|
Lower TileLang's cross-core wait flag for modes 0, 1, 2, and 4. |
Module Contents¶
- tilelang.contrib.ptodsl.sync.ascend_cross_core_set_flag(mode_id, pipe, flag_id)¶
Lower TileLang’s cross-core set flag for modes 0, 1, 2, and 4.
- tilelang.contrib.ptodsl.sync.ascend_cross_core_wait_flag(mode_id, pipe, flag_id)¶
Lower TileLang’s cross-core wait flag for modes 0, 1, 2, and 4.