vvchernov commented on code in PR #14008:
URL: https://github.com/apache/tvm/pull/14008#discussion_r1113842850
##########
python/tvm/topi/cuda/scatter_elements.py:
##########
@@ -20,6 +20,188 @@
from tvm import te, tir
from ..utils import ceil_div, get_const_int
from ..math import cast
+from .nms import atomic_add
+
+
+def gen_scatter_add_1d_atomic(data, indices, updates, out, axis, _):
Review Comment:
Yes, potentially we can. The problem is that quick search did not give me
any atomic operation like atomic_add, moreover it is not supported by some
architectures ("vulkan", "metal"). I postponed this task due to it requires
more time for investigation and implementation but does not have any priority.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
