Commit Graph

4 Commits (87852616aaf2517567a68d6b7dd5a61ab3857380)

Author SHA1 Message Date
Qi Li 946dbdae8c
[ROCM] update fluid operators for rocm (part6), test=develop (#31301)
4 years ago
wuhuanzhou e7ac74c85b
optimize compilation time of argmin/argmax op (#29595)
5 years ago
wawltor 6b28456ed0
add the argmax, argmin for the api2.0
5 years ago
Zhaolong Xing f00f982a02
add cub impl for arg max, min (#25941)
5 years ago