zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
Tree: 24ac2382ff
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '24ac2382ff'
${ noResults }
mindspore2022/mindspore/ccsrc/runtime/hardware/gpu/optimizer.h

58 lines
2.9 kB
Raw Blame History 

  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. 

  17. #ifndef MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_

  18. #define MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_

  19. 

  20. #include "backend/optimizer/common/helper.h"

  21. #include "backend/optimizer/common/optimizer.h"

  22. #include "backend/optimizer/common/pass_manager.h"

  23. #include "backend/optimizer/common/common_backend_optimization.h"

  24. #include "backend/optimizer/gpu/adam_weight_decay_fusion.h"

  25. #include "backend/optimizer/gpu/adam_fusion.h"

  26. #include "backend/optimizer/gpu/alltoall_fusion.h"

  27. #include "backend/optimizer/gpu/apply_momentum_weight_scale_fusion.h"

  28. #include "backend/optimizer/gpu/apply_momentum_scale_fusion.h"

  29. #include "backend/optimizer/gpu/apply_momentum_weight_fusion.h"

  30. #include "backend/optimizer/gpu/batch_norm_relu_fusion.h"

  31. #include "backend/optimizer/gpu/batch_norm_relu_grad_fusion.h"

  32. #include "backend/optimizer/gpu/batch_norm_add_relu_fusion.h"

  33. #include "backend/optimizer/gpu/post_batch_norm_add_relu_fusion.h"

  34. #include "backend/optimizer/gpu/batch_norm_add_relu_grad_fusion.h"

  35. #include "backend/optimizer/gpu/combine_momentum_fusion.h"

  36. #include "backend/optimizer/gpu/combine_cast_fusion.h"

  37. #include "backend/optimizer/gpu/cudnn_inplace_fusion.h"

  38. #include "backend/optimizer/gpu/insert_format_transform_op.h"

  39. #include "backend/optimizer/gpu/replace_momentum_cast_fusion.h"

  40. #include "backend/optimizer/gpu/replace_addn_fusion.h"

  41. #include "backend/optimizer/gpu/print_reduce_fusion.h"

  42. #include "backend/optimizer/gpu/remove_format_transform_pair.h"

  43. #include "backend/optimizer/gpu/remove_redundant_format_transform.h"

  44. #include "backend/optimizer/gpu/reduce_precision_fusion.h"

  45. #include "backend/optimizer/gpu/relu_v2_pass.h"

  46. #include "backend/optimizer/gpu/add_relu_v2_fusion.h"

  47. #include "backend/optimizer/gpu/add_relu_grad_v2_fusion.h"

  48. #include "backend/optimizer/graph_kernel/graph_kernel_optimization.h"

  49. #include "backend/optimizer/pass/communication_op_fusion.h"

  50. #include "backend/optimizer/gpu/concat_outputs_for_all_gather.h"

  51. #include "backend/optimizer/pass/getitem_tuple.h"

  52. #include "backend/optimizer/gpu/matmul_biasadd_fusion.h"

  53. #include "backend/optimizer/gpu/bce_with_logits_loss_fusion.h"

  54. #include "backend/optimizer/gpu/insert_cast_gpu.h"

  55. #include "backend/optimizer/gpu/adjust_depend_for_parallel_optimizer_recompute_all_gather_fusion.h"

  56. 

  57. #endif  // MINDSPORE_CCSRC_RUNTIME_HARDWARE_GPU_OPTIMIZER_H_