• A
    adds new CPU kernel for SGD op supporting BF16 data type (#32162) · 3ac6c189
    Adam Osewski 提交于
    * Initial draft for SGD BG16 kernel.
    
    * Unit tests for SGD with BF16 data type.
    
    * Add VLOG message to SGD BF16 op CPU kernel.
    
    * Enhance error messages and error types.
    
    * Refactor SGD op kernels to leverage some common code.
    
    * Make easier to add new kerne invoke code.
    
    * Fix SGD op kernel for sparse grad.
    
    * Unify quotes style.
    
    * Fix error for ROCM compilation.
    
    * Use specialized PADDLE_ENFORCE_xx functions.
    3ac6c189
sgd_op.cc 5.5 KB