Add gpu implementation of shuffle_batch_op (#33938)
* add gpu implementation of shuffle batch test=develop * add thrust cuda patches test=develop * fix macro guard * fix shuffle batch compile on windows/hip * fix hip compilation error * refine CMakeLists.txt * fix windows compile error * try to fix windows CI compilation error * fix windows compilation again * fix shuffle_batch op test on Windows
Showing
cmake/thrust.cmake
0 → 100644
patches/thrust/thrust/shuffle.h
0 → 100644
想要评论请 注册 或 登录