• Y
    [AMP OP&Test] arange op support fp16/bf16 (#51106) · f3448977
    yangjianfengo1 提交于
    * AMP arange & Test
    
    * fix arange bfloat16 dtype
    
    * update for review
    
    * update for review2
    
    * fix tile
    
    * update
    
    * fix ci
    
    * r
    
    * f
    
    * fix windows ci
    
    * update bfloat data
    
    * fix bloat16 input
    
    * add print
    
    * Update test_where_op.py
    
    * update kernel
    
    * del repeat
    
    * update review
    f3448977
arange_kernel.cu 3.6 KB