[TOPI] Make cumsum IR reusable, add thrust scan - #7303

Merged

masahi merged 15 commits into

apache:mainfrom

masahi:cuda-ex-scan

Jan 20, 2021

Commits on Jan 20, 2021

import changes from scan branch

masahi
authored and
masa
committed
d269839
fix for 1d scan

masahi
authored and
masa
committed
a7772a5
rename

masahi
authored and
masa
committed
7608e10
cast to out dtype

masahi
authored and
masa
committed
b72840e
do not run return reduction for inclusive scan

masahi
authored and
masa
committed
f2667e3
remove another ceil_div definition

masahi
authored and
masa
committed
b1bbedf
adding scan test
masa
committed
9783462
add scheduling for scan op, fixed scan 1d test
masa
committed
55df6d4
pylint fix
masa
committed
f740595
add doc string

masahi
authored and
masa
committed
b4795ef
add more thrust scan test

masahi
authored and
masa
committed
6c70ed2
add dynamic get valid count test, including empty size tensor
masa
committed
a6c7403
fix hard coded gpu targets for cpu only env
masahi
committed
e2df3c6
try retunring early if scan_size is 0
masahi
committed
a88e53c
another change for empty tensor and thrust path
masahi
committed
717270b