Improve elementwise performance. (#23001)

Browse Source

* Improve elementwise performance.

Elementwise performace is poor as walk into CommonGradBroadcastCUDA, add some new kernels for different data pattern.

* Add some cuda kernel to speedup common broadcast cases. test=develop

* Add more test cases and fix cuda kernel bug. test=develop

* Remove tests as cpu percision fails.test=develop

* Refine SplitDims, test=develop

* Change file mode, test=develop

revert-23830-2.0-beta

zhaoyuchen2018 5 years ago committed by GitHub

parent c524b930e7

commit 58615a6272

No known key found for this signature in database

GPG Key ID: 4AEE18F83AFDEB23

1 changed files with 398 additions and 9 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Show Stats Download Patch File Download Diff File

407

paddle/fluid/operators/elementwise/elementwise_op_function.h

View File

File diff suppressed because it is too large Load Diff

Write Preview

Loading…

Cancel

Save