• Z
    Add gpu implementation of shuffle_batch_op (#33938) · c6b6ba1f
    Zeng Jinle 提交于
    * 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
    c6b6ba1f
shuffle_batch_op.cu 6.0 KB