Skip to content
Snippets Groups Projects
Commit 9e1966b7 authored by Leyuan Wang's avatar Leyuan Wang Committed by Tianqi Chen
Browse files

barrier fence added for warp mem (#1174)

parent f7d9d7e8
No related branches found
No related tags found
No related merge requests found
......@@ -454,6 +454,7 @@ def build(sch,
if current_build_config().detect_global_barrier:
func = ir_pass.ThreadSync(func, "global")
func = ir_pass.ThreadSync(func, "shared")
func = ir_pass.ThreadSync(func, "warp")
warp_size = target.thread_warp_size
func = ir_pass.LowerThreadAllreduce(func, warp_size)
fsplits = [s for s in ir_pass.SplitHostDevice(func)]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment