- 09 2月, 2022 10 次提交
-
-
由 Yiqun Liu 提交于
-
由 hong 提交于
* add trace op * bug fix * bug fix; test=develop * thrust bug fix; test=develop * remove useless register; test=develop * fix bug; test=develop * update trace kernel; test=develop * move kernel args to trace_sig; test=develop
-
由 Chen Weihang 提交于
* fix slice bug of cusstom op * add offset in check
-
由 Leo Chen 提交于
-
由 Chen Weihang 提交于
-
由 hong 提交于
* update basic infrastructure; support string, suport vecotr<int>, add tensor args type index; test=develop * remove useless code; test=develop * fix bug; test=develop * polish code; test=develop
-
由 Zhang Zheng 提交于
-
由 niuliling123 提交于
-
由 zhangkaihuo 提交于
* implement AllocateFrom * dense_to_sparse_coo * optimize unit testing; support rocm * 1. delete fluid related header file 2. update the copyright * fix hipMemcpy * update dense_to_sparsecoo * add namespace sparse * sparse_csr_to_dense * test to_sparse_coo: csr_to_coo * fix writing error * to_sparse_csr: dense_to_sparse_csr and sparse_coo_to_csr * fix check shape * fix unit test * replace CUDADeviceContext by GPUContext
-
由 hong 提交于
* add norm cpu * update code; * norm bug fix * move norm op to pten; test=develop * move norm op to pten; test=develop * add norm util; test=develop * fix norm npu bug; test=develop * fix norm kernel bug; test=develop * move kernel args to pten; test=develop * move kernel args to pten sig; test=develop
-
- 08 2月, 2022 7 次提交
-
-
由 zhangbo9674 提交于
* add concat & split * add concat kernel * add concat unittest * add split unittest
-
由 Wilber 提交于
* gpu_context.. * update * update * update
-
由 niuliling123 提交于
* Replace clip, bce_loss, full and full_like with elementwise
-
由 Chen Weihang 提交于
* adapt selectedrows in execution * impl selected rows branch * support selectedrow in infershape utils * fix device compile failed * fix new exe test failed * revert some changes
-
由 From00 提交于
* Rough implementation for experiment * Support allocate cuda managed memory * Fix CI error * Modify UT * Check whether support memory oversubscription * Fix ROCM Compile error * Fix ROCM Compile error * Fix UT cuda_managed_memory_test * Set UT timeout to 40 * Add UT OOMExceptionTest * Set UT timeout to 50
-
由 Chen Weihang 提交于
* fix pten reduce dispatch bug * add cast beforce reduce * fix test failed
-
由 Leo Chen 提交于
-
- 07 2月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* refactor custom op kernel func and utils * add output sync * adapte tensor* in utils * fix windows symbol error
-
- 06 2月, 2022 1 次提交
-
-
由 Wilber 提交于
-
- 04 2月, 2022 2 次提交
-
-
由 zyfncg 提交于
* add data_transform in pten api * support GetKernelTypeForVar * fix complie problem of bfloat16 * change error namespace * add complex type transform unittest * fix merge conflict
-
由 Chen Weihang 提交于
-
- 02 2月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* remove kernel alias name * fix depreacted error * fix deprecated failed * fix mean error * resolve conflict * fix windows failed
-
- 30 1月, 2022 4 次提交
-
-
由 zhangkaihuo 提交于
* dense_to_sparse_coo * optimize unit testing; support rocm * 1. delete fluid related header file 2. update the copyright * fix hipMemcpy * update dense_to_sparsecoo * add namespace sparse * sparse_csr_to_dense * test to_sparse_coo: csr_to_coo * fix writing error
-
由 Chen Weihang 提交于
* change unary infermeta * change other infermeta * change all infermeta format * resolve conflit * fix test failed * resolve reshape conflit * fix compile failed * adapt auto api gen * fix reshape failed * fix concat failed * resolve conflict
-
由 zhangkaihuo 提交于
* dense_to_sparse_coo * optimize unit testing; support rocm * 1. delete fluid related header file 2. update the copyright * fix hipMemcpy * update dense_to_sparsecoo * add namespace sparse
-
由 Leo Chen 提交于
* upgrade _get_all_register_op_kernels * add ut * support xpu/npu * fix device id * enhance TransToFluidPlace * fix compile
-
- 29 1月, 2022 2 次提交
-
-
由 Chen Weihang 提交于
-
由 Chen Weihang 提交于
* open header for custom kernel * add core utils * tidy core code * tify header * tidy include * tidy namespace * resolve conflit * fix unittest and coverage * remove platform using * resolve conflict * resolve conflict * fix digamma namespace error * fix xpu full kernel error * fix xpu full kernel error * polish details * add place for lib storage
-
- 28 1月, 2022 5 次提交
-
-
由 Chen Weihang 提交于
* update forward argument mapping * fix compile failed * fix test failed
-
由 YuanRisheng 提交于
* refactor scale kernel that its input is selected_rows * complement upload file
-
由 hong 提交于
* move digamma to pten; test=develop * fix mutable_data bugs; test=develop * remove useless code; test=develop * remove kernel compute; test=develop * fix bug; test=develop
-
由 zyfncg 提交于
* remove remake densetensor * fix eager test error * fix bug in eager * implement AllocateFrom * remove WriteBackOutput * fix problem of eager Co-authored-by: Nzkh2016 <zhangkaihuo@baidu.com>
-
由 zyfncg 提交于
-
- 27 1月, 2022 7 次提交
-
-
由 zhangkaihuo 提交于
-
由 Chen Weihang 提交于
* add constructor for win * change impl * fix bug
-
由 zyfncg 提交于
* remove remake densetensor * fix eager test error * fix bug in eager
-
由 YuanRisheng 提交于
-
由 Aurelius84 提交于
* Support allocate_from in Tensor and allocate_data in Context * fix #ifdef CUDA * fix cycle depends * fix test_xxx_dev_api failed * fix windows compiling error * fix unittest * modify into PImpl * fix selected rows * add TODO comment * refine interface according reviewer
-
由 Chen Weihang 提交于
* add infermeta registry * add infermeta registry * add unittest * polish details
-
由 Aganlengzi 提交于
* [Demo] custom kernel based on pten kernel * merge and npu custom work well * del comments * delete other code * fix CUDAContext * fix not found small_vector.h * support NPU * fix NPUContext * fix DeviceContext support * add UT * fix call * add UT * fix * fix for comments and ut * add MACRO control * fix multi input output * support env CUSTOM_DEVICE_ROOT * deal with special cases * fix for Windows * try coverage with test_custom_kernel_dot.py * fix test_custom_kernel_dot * fix test_custom_kernel_dot * fix merge * fix merge * fix CI * update * merge and fix * remove WITH_CUSTOM_KERNEL * fix merge * merge and fix * fix ut * fix ut for mac * add more UT * add more UT * fix
-