1. 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
  2. 23 10月, 2022 1 次提交
  3. 27 9月, 2022 1 次提交
  4. 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
  5. 05 6月, 2022 1 次提交
    • S
      【code format check upgrade】 step2:yapf (#42944) · a072fca8
      Sing_chan 提交于
      * use yapf to format all python file
      
      * yapf exclude two unittests file for they rely on writing and reading file, and format will break them
      
      * disable diff_py_file because too many diff files cause command following failed
      a072fca8
  6. 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
  7. 26 4月, 2022 1 次提交