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

[FRONTEND] Added xor_sum

Browse Source
This commit is contained in:
Philippe Tillet
2021-12-16 17:55:35 -08:00
parent e575ae3443
commit 558555630f
6 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/src/triton.cc
View File

@@ -520,6 +520,7 @@ void init_triton_frontend(py::module &&m) {
m.def("min", &ir::dispatch::min, ret::reference);
m.def("max", &ir::dispatch::max, ret::reference);
m.def("sum", &ir::dispatch::sum, ret::reference);
m.def("xor_sum", &ir::dispatch::xor_sum, ret::reference);
// math
m.def("umulhi", &ir::dispatch::umulhi, ret::reference);
m.def("exp", &ir::dispatch::exp, ret::reference);

5
python/triton/language/core.py
View File

@@ -719,6 +719,11 @@ def min(input, axis, _builder=None):
def sum(input, axis, _builder=None):
return frontend.sum(input, axis, _builder)
@builtin
@_add_reduction_docstr("xor sum")
def xor_sum(input, axis, _builder=None):
return frontend.xor_sum(input, axis, _builder)
# -----------------------
# Internal for debugging