1. 26 5月, 2023 1 次提交
  2. 13 4月, 2023 1 次提交
    • H
      [enforce.h Decouple logging.h] Delete glog/logging.h from enforce.h (#52651) · 5664ea26
      HongyuJia 提交于
      * [enforce.h Decouple logging.h] Delete glog/logging.h from enforce.h
      
      * Add logging.h for profiler.cc
      
      * Add logging.h for gloo_utils.h
      
      * Add logging.h for addmm_kernel_impl.h
      
      * Add logging.h for addmm_grad_kernel_impl.h
      
      * Add logging.h for p_send_kernel.cu
      
      * Add logging.h for determinant_grad_kernel_impl.h
      
      * Add logging.h for p_recv_kernel.cu
      
      * Add logging.h for elementwise_grad_base.h
      
      * Add logging.h for transfer_layout_kernel.cc
      
      * Add logging.h for eigvals_kernel.cc and index_select_impl.h
      
      * Add logging.h for all files in kernel directory
      
      * Add logging.h for xpu_info.cc
      
      * Add logging.h for xpu
      5664ea26
  3. 06 2月, 2023 1 次提交
  4. 12 1月, 2023 1 次提交
  5. 01 11月, 2022 1 次提交
  6. 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
  7. 30 8月, 2022 1 次提交
  8. 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
  9. 31 5月, 2022 1 次提交
  10. 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
  11. 10 5月, 2022 1 次提交
  12. 26 4月, 2022 1 次提交