1. 17 3月, 2022 6 次提交
    • Y
      [fleet executor] fleet executor for npu (#40607) · 81848fff
      Yuang Liu 提交于
      81848fff
    • B
      support gpu mixed precision inference (#40531) · 06fee998
      baoachun 提交于
      06fee998
    • W
      [Eager Grad] Support eager grad interface (#40170) · 4db8cf24
      Weilong Wu 提交于
      * [Eager] Support eager grad interface, draft version
      
      * Support eager grad interface with allow_unused and multi startup_op
      
      * Fix code format
      
      * Fix allow_unused case, return PyNone if tensor not initialize
      
      * Support output's stop_gradient related to create_graph
      
      * Support grad exception case in eager mode, fix coverage CI
      
      * Update ToPyObject, return PyNone if not initialize
      
      * AccumulationNode add FLAGS_retain_grad_for_all_tensor
      
      * Fix ci issue
      
      * Fix CI issue
      
      * fix, use core.eager.Tensor
      
      * Add func SetBufferSlotRankZeros for GradTensorHolder
      
      * Support retain_graph by using ClearTensorWrappers
      
      * Support retain_graph by using ClearTensorWrappers
      
      * Update retain_graph and no_grad_vars related test case
      
      * Update code gen logic for ClearTensorWrappers
      
      * Fix by override statement
      
      * fix override func args
      
      * Support retain_graph, update unit tests
      
      * Updated ClearTensorWrappers logic
      
      * fix grad python interface
      
      * Use deep copy and update unit tests
      
      * Polish code
      
      * Polish code
      
      * Fix CI issue, Deep copy only use when user set grad_tensors
      
      * Fix CI, use Backward instead RunBackward
      
      * Fix CI, Declare kernel explicitly in test file
      
      * Polish, remove vector of TensorWrapper
      
      * Refactor the logic of grad/backward, polish codes
      
      * Update code after merge upstream develop
      
      * Polish after merge upstream develop
      
      * Update to adapt new GradNodeBase superclass
      
      * Fix error introduced during conflict resolution
      
      * Update purify potential_startup_nodes logic
      
      * Fix errors
      
      * Polish code
      
      * Remove useless args for ToPyObject
      
      * Remove useless TensorWrappersSet
      
      * Fix code-format, re-install pre-commit
      
      * Fix pre-process logic for potential_startup_ops
      
      * Update unit tests, use eager mode
      4db8cf24
    • J
      fix copy_ problem by doing it with phi copy (#40521) · c1931beb
      Jiabin Yang 提交于
      * fix copy_ problem by doing it with phi copy
      
      * improve test coverage
      
      * refactor copy with sr kernel
      c1931beb
    • C
      move grid sample op infershape (#40625) · b1b24463
      Chen Weihang 提交于
      b1b24463
    • L
      Improve the performance of fake quantize OP (#40491) · 827b6a0e
      Leo Chen 提交于
      * Move the computation of moving average scale to device
      
      * Use register to save local maximum in a thread
      827b6a0e
  2. 16 3月, 2022 21 次提交
  3. 15 3月, 2022 13 次提交
    • C
      [Phi] Move determinant op kernel into phi (#40539) · a04a6bd5
      Chen Weihang 提交于
      * add determinant phi kernel
      
      * remove original determinant op kernel
      
      * add determinant grad [hi kernel
      
      * fix determinant test failed
      
      * remove original determinant grad op kernel
      a04a6bd5
    • L
      [phi] modify the shape OP and move inferMeta of shape,matrix_pow,multi_dot (#40506) · 31729a62
      Liu-xiandong 提交于
      * [phi] move matrix_power op
      
      * MatrixInverse fluid -> phi
      
      * modify the CMake to fix compile bug
      
      * delete useless comment
      
      * mutable memory -> phi Alloc
      
      * modify the include file
      
      * modify the include file
      
      * fix bug in CI compiler
      
      * [phi]modify the shape OP and move inferMeta of shape,matrix_pow,multi_dot
      
      * delete useless comment
      
      * fix bug in CI
      
      * modify after review
      31729a62
    • R
      add number count op (#39224) · 9bdee437
      Roc 提交于
      * add expert count op
      
      add ut for expert_count
      
      * update UT only for cuda
      
      * fix for rocm
      
      * update ut
      
      * add moe module
      
      * add expert count op
      
      add ut for expert_count
      
      * update UT only for cuda
      
      * update ut
      
      * add moe module
      
      * make expert count private
      
      * rename expert count op
      Co-authored-by: Nhlygit66666 <2570058140@qq.com>
      9bdee437
    • X
      run python api in eager model and filter the out in argument list (#40523) · 4d886f75
      xiongkun 提交于
      * run python api in eager model and filter the out in argument list
      
      * fix code
      4d886f75
    • Z
      Fixed issues with generated scale operator (#40482) · 30417999
      Zhanlue Yang 提交于
      * Fixed issues with generated scale operator
      
      * Fixed minor issues
      30417999
    • F
      [NPU] add AMP O1 support (#40362) · 69dd43d1
      furnace 提交于
      * [NPU] add AMP O1 support
      
      * [NPU] fix NOTE and warnings
      69dd43d1
    • C
      [Phi] Move gather op kernel into phi (#40500) · 0c703fe7
      Chen Weihang 提交于
      * add phi gather kernel
      
      * update year
      
      * remove original gather opkernel
      
      * add gather grad phi kernels
      
      * remove origin gather grad kernel
      
      * fix failed npu and xpu
      
      * fix xpu compile failed
      0c703fe7
    • J
      oneDNN NHWC fixes (#40049) · dde9cec0
      Jacek Czaja 提交于
      * - Prototype of third solution
      
      - fix
      
      - compilation fixes
      
      - fix
      
      - fixe
      
      - fix
      
      - fix
      
      - compilation fix
      
      - comment fix
      
      - lint
      
      update mkldnn conv_elementwise_add_fuse_pass ut
      
      - NHWC changes to prelu
      
      - alhpa dims
      
      - UT fix
      
      - fix to UT
      
      - lint
      
      - Some fixes
      
      - added to BWD of prelu NHWC support
      
      - reverted removal of resetting cu_layout in clearing of caching
      
      * - Small changes
      
      * - compilation fix
      
      * - fix
      
      * - fix
      
      * lint
      
      * - fixes after internal review
      
      * - compilation fix
      
      * - lint
      dde9cec0
    • T
      add shard_id (#40261) · 6b7d4845
      Thunderbrook 提交于
      * shard_id
      
      * format
      6b7d4845
    • X
      [phi] Transfer lgamma, kldiv_loss, isclose, cumprod kernels into phi and pass... · 64223620
      xiongkun 提交于
      [phi] Transfer lgamma, kldiv_loss, isclose, cumprod kernels into phi and pass the tests of these four kernels (#39770)
      
      * tranfer and pass the lgamma unittest
      
      * merge and pass the test
      
      * transfer kldiv_loss and kldiv_loss_grad; pass the unitest
      
      * trafer the isclose and cumprod kernel
      
      * change PT_REGISTER -> PD_REGISTER
      
      * fix by code review
      
      * fix by code review
      
      * fix
      
      * remove enforce include dependence from scalar
      
      * fix
      
      * fix by code review
      
      * fix by code review
      64223620
    • C
      [Phi]move reduce_min/any/all kernel (#40374) · c46e661d
      chentianyu03 提交于
      * add reduce_min kernel
      
      * remove raw reduce_min kernel
      
      * add reduce min
      
      * add reduce any all impl
      
      * add bool reduce Kernel
      
      * remove raw any/all kernel
      
      * add any all kernel
      
      * rm comment
      c46e661d
    • Z
      Added more profile signposts to dygraph (#40201) · 36db75b4
      Zhanlue Yang 提交于
      * Added more signposts to dygraph profiling
      
      * Fixed minor issues
      
      * Refactored signpost names
      
      * Fixed typo
      
      * Removed debug codes
      
      * Fixed typo
      
      * Adjusted signpost names
      
      * Fixed issues from branch merge
      36db75b4
    • H
      Move one hot to phi (#39876) · 7701db37
      hong 提交于
      * move one hot to phi; test=develop
      
      * fix bugs; test=develop
      
      * fix bugs; test=develop
      
      * add infer meta; test=develop
      
      * fix bugs; test=develop
      
      * resolve confilct
      
      * resolve confilct
      
      * fix bug;
      
      * fix error; test=develop
      
      * update; test=develop
      
      * polish code; test=develop
      
      * add one api in eager mode; test=develop
      
      * add one hot test; test=develop
      
      * remove use less code; test=develop
      
      * fix bug; test=develop
      
      * polish code; test=develop
      
      * polish code; test=develop
      7701db37