Commit Graph

19 Commits (f1a9efcac5743e05fc4e1dbc53c08f69ec19d5a8)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
Yu Yang aadb098138 Add `op::math::set_constant` without template
7 years ago
dangqingqing ac3370a467 Add unit testing for gemv and fix the gradien check for bais.
7 years ago
qijun ab5dc9fe18 remove SelectedRows functors to selected_rows_functor.h
7 years ago
qijun f59a7c1d36 add gpu functor for SelectedRows
7 years ago
qijun 4741266d6f remove unused method
7 years ago
qijun 931572e210 SelectedRowsAddTensor method
7 years ago
qijun 5be10872f9 add selected_rows add cpu functor
7 years ago
QI JUN 2603cb7e86 Unify CUDA stream in Tensor CopyFrom interface (#4692)
7 years ago
Yi Wang 4558807c48 Use PADDLE_WITH_CUDA instead of PADDLE_WITH_GPU
7 years ago
Yu Yang 84500f9487 Change `PADDLE_ONLY_CPU` to `PADDLE_WITH_GPU`
7 years ago
qijun c634a8480a add SetConstant method in math_function.h
8 years ago
guosheng 9ffa79cd91 Add gemm with stride
8 years ago
Yu Yang 81d56ca86b Remove lazy-initialization in device_context
8 years ago
qijun c41862d2f5 merge baidu/develop
8 years ago
qijun c2631ebf6f add unittest
8 years ago
qijun 688c43b104 format code
8 years ago
qijun 5f1081d83d fix bug in dynload
8 years ago
qijun c5a7471e93 add math_function_test
8 years ago