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

1
python/src/triton.cc
View File

@@ -134,6 +134,7 @@ void init_triton_frontend(py::module &&m) {
m.def("store", &ir::dispatch::store, ret::reference);
m.def("atomic_cas", &ir::dispatch::atomic_cas, ret::reference);
m.def("atomic_xchg", &ir::dispatch::atomic_xchg, ret::reference);
m.def("atomic_add", &ir::dispatch::atomic_add, ret::reference);
// linear algebra
m.def("dot", &ir::dispatch::dot, ret::reference);
// indexing