!20466 GPU codex fix

Merge pull request !20466 from VectorSL/code-check
This commit is contained in:
i-robot 2021-07-22 09:01:22 +00:00 committed by Gitee
commit ec7ce3494a
5 changed files with 1 additions and 19 deletions

View File

@ -14,10 +14,6 @@
* limitations under the License.
*/
#include "backend/optimizer/gpu/combine_cast_fusion.h"
#include <vector>
#include <string>
#include "backend/session/anf_runtime_algorithm.h"
#include "ir/primitive.h"
#include "utils/utils.h"

View File

@ -14,11 +14,6 @@
* limitations under the License.
*/
#include "backend/optimizer/gpu/combine_momentum_fusion.h"
#include <memory>
#include <vector>
#include <string>
#include "backend/session/anf_runtime_algorithm.h"
#include "ir/primitive.h"
#include "utils/utils.h"

View File

@ -14,11 +14,6 @@
* limitations under the License.
*/
#include "backend/optimizer/gpu/replace_addn_fusion.h"
#include <memory>
#include <vector>
#include <string>
#include "backend/session/anf_runtime_algorithm.h"
#include "ir/primitive.h"
#include "utils/utils.h"

View File

@ -14,11 +14,6 @@
* limitations under the License.
*/
#include "backend/optimizer/gpu/replace_momentum_cast_fusion.h"
#include <memory>
#include <vector>
#include <string>
#include "backend/session/anf_runtime_algorithm.h"
#include "ir/primitive.h"
#include "utils/utils.h"

View File

@ -384,6 +384,7 @@ class MakeRefKey(Primitive):
def __call__(self):
pass
class FusedWeightScaleApplyMomentum(PrimitiveWithInfer):
"""
Optimizer that implements the Momentum algorithm with weight decay and loss scale.