Files
apache--tvm/python
zhuwenxi d043cb962f [BugFix][TOPI] Fix the integer overflow problem of the scatter_nd op. (#8415)
* Fix the integer overflow problem of the scatter_nd op.

* Fix scatter_nd's crash problem:
1. Existing scatter_nd cuda implementation has a very large bound,
   which could overflow int32 range when input tensor shape is
   large enough;
2. The overflow could cause the if statement always evaluate to
   true, thus conducts invalid memory accesses;
3. We fix this problem in this commit by reducing the bound, the
   original large bound is not only unnecessary, but also degrading
   the performance; With this fix, scatter_op's performance improves
   100x on some cases.

Co-authored-by: wenxizhu <wenxizhu@tencent.com>
2021-07-12 20:57:16 -06:00
..