Commit 37f745f

authored and

zhaoyingli

committed

add adamw cuda kernel (PaddlePaddle#35020)

* adamw support cuda * adamw support cuda

1 parent 9750eb1 commit 37f745fCopy full SHA for 37f745f

5 files changed

+555

-98

lines changed

paddle/fluid
- operators/optimizers
  - adamw_op.cu
  - adamw_op.h
- pybind
  - op_function_generator.cc
python/paddle
- fluid/tests/unittests
  - test_adamw_op.py
- optimizer
  - adamw.py

+555

-98

lines changed

Comments

(0)