You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

optimizer.h 2.6 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. /**
  2. * Copyright 2021 Huawei Technologies Co., Ltd
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #ifndef MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_
  17. #define MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_
  18. #include "backend/optimizer/common/helper.h"
  19. #include "backend/optimizer/common/optimizer.h"
  20. #include "backend/optimizer/common/pass_manager.h"
  21. #include "backend/optimizer/common/common_backend_optimization.h"
  22. #include "backend/optimizer/gpu/adam_weight_decay_fusion.h"
  23. #include "backend/optimizer/gpu/adam_fusion.h"
  24. #include "backend/optimizer/gpu/apply_momentum_weight_scale_fusion.h"
  25. #include "backend/optimizer/gpu/apply_momentum_scale_fusion.h"
  26. #include "backend/optimizer/gpu/apply_momentum_weight_fusion.h"
  27. #include "backend/optimizer/gpu/batch_norm_relu_fusion.h"
  28. #include "backend/optimizer/gpu/batch_norm_relu_grad_fusion.h"
  29. #include "backend/optimizer/gpu/batch_norm_add_relu_fusion.h"
  30. #include "backend/optimizer/gpu/post_batch_norm_add_relu_fusion.h"
  31. #include "backend/optimizer/gpu/batch_norm_add_relu_grad_fusion.h"
  32. #include "backend/optimizer/gpu/combine_momentum_fusion.h"
  33. #include "backend/optimizer/gpu/combine_cast_fusion.h"
  34. #include "backend/optimizer/gpu/cudnn_inplace_fusion.h"
  35. #include "backend/optimizer/gpu/insert_format_transform_op.h"
  36. #include "backend/optimizer/gpu/replace_momentum_cast_fusion.h"
  37. #include "backend/optimizer/gpu/replace_addn_fusion.h"
  38. #include "backend/optimizer/gpu/print_reduce_fusion.h"
  39. #include "backend/optimizer/gpu/remove_format_transform_pair.h"
  40. #include "backend/optimizer/gpu/remove_redundant_format_transform.h"
  41. #include "backend/optimizer/gpu/reduce_precision_fusion.h"
  42. #include "backend/optimizer/gpu/relu_v2_pass.h"
  43. #include "backend/optimizer/gpu/add_relu_v2_fusion.h"
  44. #include "backend/optimizer/gpu/add_relu_grad_v2_fusion.h"
  45. #include "backend/optimizer/graph_kernel/graph_kernel_optimization.h"
  46. #include "backend/optimizer/pass/communication_op_fusion.h"
  47. #include "backend/optimizer/pass/getitem_tuple.h"
  48. #endif // MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_