Commit Graph

8 Commits (develop)

Author SHA1 Message Date
Chen Weihang 878e117b6d
[CustomOp] Support float16 in custom op (#31725)
5 years ago
Chen Weihang 87852616aa
[CustomOp] Support complex dtype in custom op (#31657)
5 years ago
Aurelius84 fadabbe9b0
[CustomOp] Automatically specify PADDLE_WITH_MKLDNN & Remove Interpreter argument (#31391)
5 years ago
Jiabin Yang 0c38708a90
[Custom Op] Remove unsupport dtypes (#31232)
5 years ago
Aurelius84 912022fa0c
[CustomOp]Add cpp_extension en doc (#31187)
5 years ago
Aurelius84 406f4a7513
[CustomOp] Support to specific extra_cflags and exctra_cuda_flags independently (#31059)
5 years ago
Chen Weihang 1ce96fa118
[CustomOp] Add new paddle custom op so (#31141)
5 years ago
Chen Weihang e60fd1f6a8
[CustomOp] Split test and add inference test (#31078)
5 years ago