[TE] Support mixing normal and cross-thread reduction (#5193)
authorTang, Shizhi <beantang.tang@gmail.com>
Sat, 4 Apr 2020 01:49:56 +0000 (09:49 +0800)
committerGitHub <noreply@github.com>
Sat, 4 Apr 2020 01:49:56 +0000 (18:49 -0700)
commitb41f4e5523b1f5974a4bdfe4dcc3233903394224
tree7f75b193bba12020839adf3a87a11056ec7a0915
parent75e936e1b5db305864c76277e2ba47c453c4c6a8
[TE] Support mixing normal and cross-thread reduction (#5193)

* Support mixing normal and cross-thread reduction

* minor improvements
src/te/operation/compute_op.cc
src/te/operation/cross_thread_reduction.cc
tests/python/unittest/test_target_codegen_cuda.py