- 17 3月, 2022 5 次提交
-
-
由 Chen Weihang 提交于
-
由 Leo Chen 提交于
* Move the computation of moving average scale to device * Use register to save local maximum in a thread
-
由 Wilber 提交于
* infrt add trt engine * fix register * file generate * fix ci error * fix conflict * add copyright * update * update * update * update engine name * refactor trt code * update * update * update * update * fix conflict * update * fix compile with cuda
-
由 王明冬 提交于
-
由 王明冬 提交于
-
- 16 3月, 2022 31 次提交
-
-
由 Zhong Hui 提交于
* segment pool support for int int64 kernel. * add support in python api
-
由 Zhang Zheng 提交于
* Optimize the computation of log_softmax * modify the var name
-
由 Chen Weihang 提交于
-
由 Lijunhui 提交于
* init commit * correct namespace
-
由 huzhiqiang 提交于
-
由 From00 提交于
* Add yaml config for pool2d * Fix CI error * Fix code format error
-
由 From00 提交于
-
由 Aurelius84 提交于
* [Phi] Migrate mode_op and mode_grad_op into Phi * fix omp * add ifdef * migrate infershape * modify according reviewer
-
由 piotrekobi 提交于
* Refactor elementwise op grad classes * Add more refactor changes * Revert set layout and format deletion * Fix failing elementwise test
-
由 Zuza 提交于
* Quantize elementwise mul op * Parametrize elementwise functions * Fix code formatting
-
由 ronnywang 提交于
-
由 Aurelius84 提交于
* Fix tile_op inferShape * fix style
-
由 zlsh80826 提交于
-
由 zhangkaihuo 提交于
restructure conv
-
由 zyfncg 提交于
* move roi_align_grad kernel * move roi_align grad kernel and infershape to phi * remove roi_align infershape
-
由 chenenquan 提交于
* [PHI] Migrate roll op * 【phi】migrate eigh op to phi (#40213) * migrate eigh to phi * optimize code * modify code according to comment * conflict resolution * [PHI] Migrate roll op * [PHI] Fix converage of roll_sig * [PHI] Fix infermate of roll_sig * [Phi] Fix unittest coverage of roll op * [PHI] Fix infermeta in unary * [PHI] Fix parameter type of roll op * [PHI] Fix parameter type of roll op * [PHI] Fix parameter of roll op Co-authored-by: Ncrystal <62974595+Zjq9409@users.noreply.github.com>
-
由 chenenquan 提交于
* [PHI] Migrate index_select op * [PHI] Fix bug in test_variable * [PHI] migrate index_select op
-
由 YuanRisheng 提交于
-
由 Liu-xiandong 提交于
* [kp]fix bug that cannot fallback to CPU normally in XPU KP * fix bug in static graph
-
由 caozhou 提交于
* migrate multiplex op kernel * migrate qr cpu kernel * migrate tril_triu op kernel * fix multiplex kernel * add kernel sig * fix dependence and bug * fix multiplex error * fix npu include error * fix conflict * fix conflict and delete tril_triu * fix date and multiplex input * adapt header file order * fix header file include * fix conflict * delete cholesky_solve_op.h * delete triangular_solve_op.h
-
由 王明冬 提交于
-
由 chentianyu03 提交于
* move reduce kernels into one file * rename reduce_prod to prod * move reduce sum/mean from math_kernel into reduce_kernel * rm comment
-
由 xiongkun 提交于
-
由 Chen Weihang 提交于
-
由 Chen Weihang 提交于
* add grid sample phi kernel * add grid sample phi kernel and remove original kernel * replace mutable_data by alloc
-
由 qipengh 提交于
-
由 Zhanlue Yang 提交于
-
由 Chen Weihang 提交于
-
由 Yulong Ao 提交于
* [Auto Parallel] Support the auto completion of while_op * [Auto Parallel] Improve the completion algorithms * [Auto Parallel] Fix bugs for ernie inference * [Auto Parallel] Remove attrs which cannot be pickled * [Auto Parallel] make the dims_mappings of LodTensorArray vars empty * [Auto Parallel] Fix bugs for the ernie inference in the pipeline parallel * [Auto Parallel] Remove unncessary comments * [Auto Parallel] Fix a bug of the CMakeLists * [Auto Parallel] Use the newest APIs to write the unit test * [Auto Parallel] Remove unnecessary statements
-
由 xiongkun 提交于
-
由 王明冬 提交于
* change some symbol names * add test * add phi to opt.cc * clean code * up * update * up * up * Update pten_pass.mlir * Update convolution_grad_kernel.cc * update * restore init_infrt_dialects * restore * up * up * up Co-authored-by: NSuperjomn <yanchunwei@outlook.com>
-
- 15 3月, 2022 4 次提交
-
-
由 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
-
由 Chen Weihang 提交于
-
由 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
-
由 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>
-