Skip to content

Commit b764626

Browse files
committed
enabled sync statement to use ATMI
1 parent 49b3883 commit b764626

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

compiler/AST/build.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2940,7 +2940,11 @@ buildSyncStmt(Expr* stmt) {
29402940
block->insertAtTail(new CallExpr(PRIM_MOVE, endCountSave, new CallExpr(PRIM_GET_END_COUNT)));
29412941
block->insertAtTail(new CallExpr(PRIM_SET_END_COUNT, new CallExpr("_endCountAlloc", /* forceLocalTypes= */gFalse)));
29422942
block->insertAtTail(stmt);
2943+
#ifndef TARGET_HSA
29432944
block->insertAtTail(new CallExpr("_waitEndCount"));
2945+
#else
2946+
block->insertAtTail(new CallExpr("_finalizeTaskGroup"));
2947+
#endif
29442948
block->insertAtTail(new CallExpr("_endCountFree", new CallExpr(PRIM_GET_END_COUNT)));
29452949
block->insertAtTail(new CallExpr(PRIM_SET_END_COUNT, endCountSave));
29462950
return block;

0 commit comments

Comments
 (0)