1. 16 1月, 2023 8 次提交
  2. 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
  3. 14 1月, 2023 1 次提交
  4. 13 1月, 2023 25 次提交
  5. 12 1月, 2023 4 次提交
    • X
      fix_arg (#49770) · 5d60ff91
      xiaoxiaohehe001 提交于
      5d60ff91
    • S
      lerp support 0 Tensor (#49667) · 8cd0d5b3
      sunli 提交于
      * lerp support 0 Tensor
      
      * fix lerp grad
      
      * fix lerp zero test
      
      * fix 0D + ND/ND + 0D
      
      * fix check
      
      * update code
      
      * fix lerp infer shape
      
      * static backward test
      
      * updata static graph test
      8cd0d5b3
    • W
      Migrate collective communication checks to PHI (#49754) · c24e7fe1
      Wen Sun 提交于
      * refactor: migrate comm checks
      
      * refactor: add check in comm context
      
      * feat: add gloo static check
      
      * refactor: add place param in static check
      c24e7fe1
    • J
      Fix reduce func bug in process_group_bkcl (#49749) · 8e291bf7
      jameszhang 提交于
      * Fix reduce func bug in process_group_bkcl
      
      Also catch up with a recent process_group PR that failed to add XPU branch.
      Note that reduce is still accomplished by allreduce for xpu. Fix this should
      xccl lib be updated.
      
      * fix compile issue for non-XPU
      8e291bf7