fix code check

This commit is contained in:
VectorSL 2021-07-19 09:33:08 +08:00
parent c9af7643d7
commit 089d8b37e5
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.