Skip to content

Commit d288bbc

Browse files
Matthew BrookhartMasahiro Masuda
andauthored
[TOPI][GPU] Mergepath sort with odd-even block sort (#7611)
* Mergepath sort with odd-even block sort * fix lint, add test * respond to review comments * speed up tests by reducing dtype skews * fix bad rebase * change threading to support vulkan * fix lint * only sort if the data is non-empty * fix lint again * fix for vk * move if to higher scope * fix typo Co-authored-by: Masahiro Masuda <masahi@[email protected]>
1 parent 348d4e7 commit d288bbc

File tree

4 files changed

+457
-164
lines changed

4 files changed

+457
-164
lines changed

include/tvm/tir/stmt.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1312,6 +1312,10 @@ constexpr const char* fragment_shape = "fragment_shape";
13121312
*/
13131313
constexpr const char* fragment_layout = "fragment_layout";
13141314

1315+
/*!
1316+
* \brief Mark that the kernel is hand threaded and doesn't need syncs inserted
1317+
*/
1318+
constexpr const char* hand_threaded = "hand_threaded";
13151319
/*!
13161320
* \brief Check if attr_key is a pragma key extension
13171321
* \param attr_key The attr key to be compared

0 commit comments

Comments
 (0)