gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PYTHON] Added atomic_add (#94)

Browse Source
This commit is contained in:
Philippe Tillet
2021-04-29 09:13:45 -04:00
committed by Philippe Tillet
parent d7f87929fa
commit 2b75158426
5 changed files with 65 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/ir/dispatch.cc
View File

@@ -521,6 +521,15 @@ ir::value *dispatch::atomic_xchg(ir::value* ptr, ir::value *val, ir::builder *bu
return builder->create_atomic_exch(ptr, val);
}
ir::value *dispatch::atomic_add(ir::value* ptr, ir::value *val, ir::value *mask, ir::builder *builder){
if(!mask){
mask = builder->get_int1(true);
if(ptr->get_type()->is_block_ty())
mask = builder->create_splat(mask, ptr->get_type()->get_block_shapes());
}
return builder->create_atomic_add(ptr, val, mask);
}
//===----------------------------------------------------------------------===//
// Linear Algebra
//===----------------------------------------------------------------------===//