1. 17 1月, 2023 1 次提交
    • X
      【Prim】Add multiply,expand,div vjp rules (#49831) · 39c6765a
      Xiaoxu Chen 提交于
      * support elementwise base func
      
      * fix compiling error and add test
      
      * support vjp for div using comp
      
      * remove additional change
      
      * fix dy2st error with magic num
      
      * fix dy magic num
      
      * another magic
      
      * another magic
      
      * another magic
      
      * add skip rename strategy
      
      * support add vjp
      
      * support add with new axis cal
      
      * support sub vjp
      
      * [prim] add multiply vjp rules
      
      * [prim] add multiply vjp rules
      
      * [prim] fix no infershape with composite in _append_backward_ops
      
      * [prim] add expand vjp rule
      
      * [prim] add exp vjp rule
      
      * uncomment infer shape for reshape/sum static prim api
      
      * [prim] fix tanh nullptr error
      
      * remove some print message
      
      * fix magic number in run_program relative tests @JiaBinYang
      
      * [prim] add expand,multiply,exp vjp rules
      
      * fix only support single direction reduce error
      
      * infer reduce dims using out dims
      Co-authored-by: NJiabinYang <360788950@qq.com>
      39c6765a
  2. 16 1月, 2023 22 次提交
  3. 15 1月, 2023 2 次提交
    • R
      support mp on xpu (#49815) · 6a56bce7
      Roc 提交于
      1 update xccl lib
      2 when using comm_ctx, the allocator should be set manually.
      6a56bce7
    • J
      【Prim】Enhance tests (#49814) · 090aa45d
      Jiabin Yang 提交于
      * support elementwise base func
      
      * fix compiling error and add test
      
      * remove additional param
      
      * support vjp for div using comp
      
      * remove additional change
      
      * fix dy2st error with magic num
      
      * fix dy magic num
      
      * another magic
      
      * another magic
      
      * add more test
      
      * fix windows problem
      
      * another magic
      
      * fix windows compile
      
      * invoke ci
      
      * add skip rename strategy
      
      * support add vjp
      
      * fix test_tanh
      
      * support add with new axis cal
      
      * fix resnet and some test
      
      * add composite log
      
      * support sub vjp
      
      * enhance_tests
      
      * support more dtype for full
      090aa45d
  4. 14 1月, 2023 1 次提交
  5. 13 1月, 2023 14 次提交