Browse Source

move add_memcpy_async pass and transdata_split pass

tags/v0.3.0-alpha
huanghui 6 years ago
parent
commit
9e29224af9
7 changed files with 12 additions and 12 deletions
  1. +2
    -2
      mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc
  2. +1
    -1
      mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc
  3. +3
    -3
      mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.h
  4. +1
    -1
      mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.cc
  5. +3
    -3
      mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.h
  6. +1
    -1
      tests/ut/cpp/pre_activate/ascend/enhancer/add_memcpy_async_test.cc
  7. +1
    -1
      tests/ut/cpp/pre_activate/ascend/ir_fission/transdata_split_test.cc

+ 2
- 2
mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.cc View File

@@ -39,7 +39,7 @@
#include "pre_activate/ascend/ir_fusion/adam_apply_one_with_decay_rule.h"
#include "pre_activate/ascend/ir_fusion/parameter_and_transop_fusion.h"
#include "pre_activate/ascend/ir_fusion/transpose_transdata_fusion.h"
#include "pre_activate/ascend/ir_fusion/transdata_split.h"
#include "pre_activate/ascend/ir_fission/transdata_split.h"
#include "pre_activate/ascend/ir_fission/topk_split.h"
#include "pre_activate/ascend/ir_fusion/momentum_lossscale_fusion.h"
#include "pre_activate/ascend/ir_fusion/mul_add_fusion.h"
@@ -59,7 +59,7 @@
#include "pre_activate/ascend/format_type/check_consistency.h"
#include "pre_activate/ascend/buffer_fusion/buffer_fusion.h"
#include "pre_activate/ascend/format_type/deal_ref_trans_and_cast.h"
#include "pre_activate/ascend/ir_fission/add_memcpy_async.h"
#include "pre_activate/ascend/enhancer/add_memcpy_async.h"
#include "pre_activate/ascend/format_type/insert_cast_for_runop.h"
#include "pre_activate/ascend/format_type/insert_transdata_for_runop.h"
#include "pre_activate/ascend/enhancer/getnext_memcpy_elimination.h"


mindspore/ccsrc/pre_activate/ascend/ir_fission/add_memcpy_async.cc → mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc View File

@@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "pre_activate/ascend/ir_fission/add_memcpy_async.h"
#include "pre_activate/ascend/enhancer/add_memcpy_async.h"
#include <vector>
#include "utils/utils.h"
#include "session/anf_runtime_algorithm.h"

mindspore/ccsrc/pre_activate/ascend/ir_fission/add_memcpy_async.h → mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.h View File

@@ -13,8 +13,8 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_ADD_MEMCPY_ASYNC_H_
#define MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_ADD_MEMCPY_ASYNC_H_
#ifndef MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_ENHANCER_ADD_MEMCPY_ASYNC_H_
#define MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_ENHANCER_ADD_MEMCPY_ASYNC_H_

#include <memory>
#include "pre_activate/common/optimizer.h"
@@ -28,4 +28,4 @@ class AddMemcpyAsync : public PatternProcessPass {
};
} // namespace opt
} // namespace mindspore
#endif // MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_ADD_MEMCPY_ASYNC_H_
#endif // MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_ENHANCER_ADD_MEMCPY_ASYNC_H_

mindspore/ccsrc/pre_activate/ascend/ir_fusion/transdata_split.cc → mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.cc View File

@@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "pre_activate/ascend/ir_fusion/transdata_split.h"
#include "pre_activate/ascend/ir_fission/transdata_split.h"
#include <set>
#include "pre_activate/ascend/ascend_helper.h"
#include "session/anf_runtime_algorithm.h"

mindspore/ccsrc/pre_activate/ascend/ir_fusion/transdata_split.h → mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.h View File

@@ -13,8 +13,8 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FUSION_TRANSDATA_SPLIT_H_
#define MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FUSION_TRANSDATA_SPLIT_H_
#ifndef MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_TRANSDATA_SPLIT_H_
#define MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_TRANSDATA_SPLIT_H_
#include <vector>
#include <string>
#include <utility>
@@ -42,4 +42,4 @@ class TransDataSplit : public Pass {
};
} // namespace opt
} // namespace mindspore
#endif // MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FUSION_TRANSDATA_SPLIT_H_
#endif // MINDSPORE_CCSRC_PRE_ACTIVATE_ASCEND_IR_FISSION_TRANSDATA_SPLIT_H_

tests/ut/cpp/pre_activate/ascend/ir_fission/add_memcpy_async_test.cc → tests/ut/cpp/pre_activate/ascend/enhancer/add_memcpy_async_test.cc View File

@@ -22,7 +22,7 @@
#include "utils/utils.h"
#include "kernel/kernel_build_info.h"
#include "pre_activate/common/optimizer.h"
#include "pre_activate/ascend/ir_fission/add_memcpy_async.h"
#include "pre_activate/ascend/enhancer/add_memcpy_async.h"

namespace mindspore {
namespace opt {

tests/ut/cpp/pre_activate/ascend/ir_fusion/transdata_split_test.cc → tests/ut/cpp/pre_activate/ascend/ir_fission/transdata_split_test.cc View File

@@ -23,7 +23,7 @@
#define private public
#define protected public
#include "pre_activate/ascend/format_type/insert_trans_op.h"
#include "pre_activate/ascend/ir_fusion/transdata_split.h"
#include "pre_activate/ascend/ir_fission/transdata_split.h"
#undef private
#undef protected


Loading…
Cancel
Save