search

ROCm / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
18 stars 14 forks source link

Add rocblas_alt_impl flag for backprop in MLP #71

Closed hubertlu-tw closed 2 years ago

hubertlu-tw commented 2 years ago

The unit tests for MLP (pytest test/L0/run_mlp/) has been tested and passed.