1. 06 2月, 2023 1 次提交
  2. 12 1月, 2023 1 次提交
  3. 01 11月, 2022 1 次提交
  4. 31 10月, 2022 1 次提交
    • X
      [Einsum] Einsum support repeated labels. (#47290) · 6e1c14e3
      xiongkun 提交于
      * add unittest for einsum-v2-trace and diagonal
      
      * repeat labels.
      
      * einsum support repeated labels.
      
      * forward is ok for diagonal and undiagonalized.
      TODO: check backward is ok by our theorem.
      
      * backward is ok!
      
      * fix by PR suggestions.
      
      * fix ci error
      
      * fix ci error
      
      * fix ci warning
      6e1c14e3
  5. 30 8月, 2022 1 次提交
  6. 14 6月, 2022 1 次提交
    • X
      [ Make FLAGS_einsum_opt as default ] Einsum memory optimization (#43397) · 83abec60
      xiongkun 提交于
      * change logic for optimize
      
      * modifty
      
      * optimize the backward speed of EinsumOp
      
      * add cache optimizer for einsum op
      
      * EinsumOp: fix new dygraph mode error
      
      * fix bug
      
      * change Cache->InnerCache
      
      * fix code
      
      * fix
      
      * add nan inf utils for einsum op
      
      * add as_extra
      
      * memory optimizer for einsum
      
      * update code
      83abec60
  7. 31 5月, 2022 1 次提交
  8. 25 5月, 2022 1 次提交
    • X
      [EinsumOp] Optimize the backward speed of EinsumOp (#42663) · 71b046cd
      xiongkun 提交于
      * change logic for optimize
      
      * modifty
      
      * optimize the backward speed of EinsumOp
      
      * add cache optimizer for einsum op
      
      * EinsumOp: fix new dygraph mode error
      
      * fix bug
      
      * change Cache->InnerCache
      
      * fix code
      
      * fix
      
      * add nan inf utils for einsum op
      
      * add as_extra
      
      * Compatible with v2.3 EinsumOp
      
      * remove dispensable
      71b046cd
  9. 10 5月, 2022 1 次提交
  10. 26 4月, 2022 1 次提交