- 19 5月, 2023 5 次提交
-
-
由 warrentdrew 提交于
* add minimum grad composite rules * add public python api * fix format * fix format * update testcase * fix testcase * fix format * fix cmakelist.txt * fix format * fix param problem * fix op and composite rule * fix bf16 cpu support problem * fix bf16 cpu issue * fix axis error log * add axis for maximum * revert commit * remove .orig * fix generic problem * revert max op * fix axis error * fix maximum axis * fix test_check_output * fix cinn * fix minimum maximum axis check
-
由 limingshu 提交于
* Reorganize the forward codes of flash-attention. * Fix forward. * Remove some noused codes. * Simplify codes and fix backward. * Change all LOG(INFO) to VLOG and fix the backward. * add scale for AF2 flash_attn, much thanks to xreki and shaojie for debug these codes * decrease the effect of debug print on performance * Unify the initialize of flashattn arguments. * Rewirte the reshape of temp_mask and temp_bias. * API support use_flash_attn. * Fix compiling error on CI. * Try to crop the flash-attention lib. * Correct the condition of whether can use flash-attn. * Remove the softmax_out argument. * Remove is_causal. * Polish codes. * Fix qkv_transpose_out's shape and scaling of Q * K. * Update commit of flash-attention. --------- Co-authored-by: NLiu Yiqun <liuyiqun01@baidu.com>
-
由 Zhang Zheng 提交于
* Add large dim test of log_softmax * fix
-
由 Charles-hit 提交于
-
由 Danyang Zhang 提交于
* delete bf16 of cross entropy * delete bf16 of cross entropy
-
- 18 5月, 2023 14 次提交
-
-
由 houj04 提交于
-
由 Charles-hit 提交于
* add meshgrid,expand_as, prod and grad bf16 kernel * fix bf16 for optest * modify code style * fix amp test
-
由 PuQing 提交于
* fix parameter not passed * fix repr
-
由 HongyuJia 提交于
* [CINN] Fix TestGelu unittest of CINN * pass if_enable_cinn
-
由 co63oc 提交于
-
由 Hulek 提交于
* Fused elementwises kernels and ops * change fuse pass name * adjust .pbtxt files * adjust quantization attributes * add missing arguments and fix others, review fixed * simplify fused kernel registration * fix elementwise unit tests * reuse one fused elementwise op * adjust proto * Add supported datatypes * Change 'Scale' to 'scale' in tests, change some tests to onednn * Revert breaking changes * Fix unit tests * Delete obsolete test cases * Delete commented out code * Fix codestyle * delete temporary condition * fix conflicts and delete duplicate fusing * Fix code after merge * Move tests to new directory * fix tests volatility * Rename test_elementwise_add_onednn_op.py to test_elementwise_add_mkldnn_op.py * Update CMakeLists.txt add mkldnn op test --------- Co-authored-by: NSilv3S <slawomir.siwek@intel.com>
-
由 co63oc 提交于
-
由 co63oc 提交于
-
由 LoneRanger 提交于
-
由 RedContritio 提交于
* simplify layer_norm_op.cc * support auto generate for op layer_norm * update unittest for composite_layer_norm * remove layer_norm_op.cc from scripts * replace layer_norm_op with generated_op * add get_expected_kernel for layer_norm * update cmake kernel register function for layer_norm_mkldnn_op
-
由 houj04 提交于
* [XPU] do not call check_nccl_version_for_p2p under xpu * refine code.
-
由 张春乔 提交于
* rm cmake npu * Update generic.cmake * Update generic.cmake
-
由 co63oc 提交于
-
由 shaojie_wang 提交于
* add master gradients on static graph * add unit test for bf16 master grad static graph * use float16 as v100 test dtype * only skip GPU which do not support bf16 * use linear layer to test master grad * 1.push master grad creation before all optimizer ops; 2.remove useless unittest; 3.use a function to create master grad states
-
- 17 5月, 2023 4 次提交
-
-
由 jiangcheng 提交于
-
由 duanyanhui 提交于
* suport device_guard for npu * fix comment * fix typo
-
由 JYChen 提交于
-
由 LyndonKong 提交于
* Add i1 and i1e op * resolve merge conflicts
-
- 16 5月, 2023 17 次提交
-
-
由 co63oc 提交于
-
由 zhouweiwei2014 提交于
-
由 Difer 提交于
* add bf16 for mode * remove random seed 666 * try to fix op_type error * test for me * try to fix op_type * fix redundancy code * add fp,bf for lastdim * fix some error * simplify code * fix shape error * optype error * fix skipif bf16
-
由 Yuang Liu 提交于
-
由 zhenhailiu 提交于
* polish * polish
-
由 xiaoguoguo626807 提交于
* add rules * modify no kernel yaml parse * success op generate * success test_silu_double * modify bug * modify static error * modify silu_grad input * modify kernel signature * modify kernel signature * code style * code style * review * delete opinfo modify * modify gradOpMaker * modify gradOpMaker * modify genarated-j2 * add approve rules * modify aytograd_functional_static_test
-
由 niuliling123 提交于
-
由 张春乔 提交于
* rm npu * rm use_npu * rm npuid * rm use_npu * rm npuid * delete npupinned * roll back sth. * roll back sth. * delete npupinned * roll back sth. * roll back sth. * rm npu * rollback something * rollback npu identity * rollback npu identity
-
由 cxxly 提交于
-
由 Sonder 提交于
* trans fused batch norm Compute function * trans batch norm register info to phi * trans fused batch norm grad Compute * trans batch norm grad register info * add sig file * update sig file * Update fused_bn_activation_kernel.cu * Update fused_bn_activation_grad_kernel.cu * fix * Rename fused_bn_activation_kernel_grad.cu to fused_bn_activation_kernel.cu * fix * fix * fix CudnnDataType error * fix * fix include * update * add #if * add fused bn act to cmakelist.txt * update cmakelist * fix #ifdef error * add timeout set * add env set * fix * fix * Update fused_bn_activation_sig.cc
-
由 cyberslack_lee 提交于
-
由 cyberslack_lee 提交于
-
由 meteor135 提交于
* remove _in_eager_mode_ * remove _in_eager_mode_
-
由 co63oc 提交于
-
由 Yiqun Liu 提交于
* Allow to switch whether to use promote strategy to choose kernel for O2 training. * Fix comparing error and add unittest.
-
由 gouzil 提交于
* [phi]mv StftKernel to phi * [phi] fix KernelSignature * [phi]fix arr error * [phi] Disable check_dygraph * [phi]fix include * [phi] rewrite mutable_data, add output register * [phi] fix Alloc * [phi] fix Alloc again * [phi] fix mutable_data * [phi] fix onesided_out Resize
-
由 Mahmoud Ashraf 提交于
* correct 1th to 1st * correct 1th to 1st * fix typo * fix typos
-