move graph_kernel & mem_reuse to backend

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2023-03-03 15:42:29 +08:00
parent 19735e8afd
commit a1a3f459f9
255 changed files with 428 additions and 424 deletions

View File

@ -390,12 +390,12 @@ set(BACKEND_SUB_COMP
ps
distributed
kernel
common/mem_reuse
backend/common/mem_reuse
backend/common/optimizer
backend/common/pass
backend/common/session
backend/common/somas
common/graph_kernel
backend/common/graph_kernel
backend/graph_compiler
backend/operator
runtime/collective

View File

@ -3,4 +3,4 @@ file(GLOB_RECURSE _GK_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
)
list(APPEND _GRAPH_KERNEL_SRC_FILES ${_GK_SRC_LIST})
add_library(_mindspore_common_graph_kernel_obj OBJECT ${_GRAPH_KERNEL_SRC_FILES})
add_library(_mindspore_backend_common_graph_kernel_obj OBJECT ${_GRAPH_KERNEL_SRC_FILES})

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/callback_impl.h"
#include "backend/common/graph_kernel/adapter/callback_impl.h"
#include <algorithm>
#include <vector>
@ -23,7 +23,7 @@
#include "backend/common/session/anf_runtime_algorithm.h"
#include "include/common/utils/anfalgo.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/adapter/fake_abstract_shape.h"
#include "backend/common/graph_kernel/adapter/fake_abstract_shape.h"
#include "kernel/graph_kernel_info.h"
namespace mindspore::graphkernel {

View File

@ -18,7 +18,7 @@
#define MINDSPORE_CCSRC_BACKEND_OPTIMIZER_GRAPH_KERNEL_ADAPTER_CALLBACK_IMPL_H_
#include <string>
#include <vector>
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
namespace mindspore::graphkernel {
class BACKEND_EXPORT CallbackImpl : public Callback {

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/adapter/expander.h"
#include "backend/common/graph_kernel/adapter/expander.h"
#include <map>
#include <set>
@ -24,13 +24,13 @@
#include <utility>
#include "include/common/utils/python_adapter.h"
#include "kernel/akg/akg_kernel_json_generator.h"
#include "common/graph_kernel/split_umonad.h"
#include "common/graph_kernel/substitute_dropout.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/adapter/callback_impl.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/graph_kernel/split_umonad.h"
#include "backend/common/graph_kernel/substitute_dropout.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/adapter/callback_impl.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/pass/inplace_assign_for_custom_op.h"
#include "kernel/common_utils.h"
#include "utils/ms_context.h"

View File

@ -16,7 +16,7 @@
#ifndef MINDSPORE_CCSRC_COMMON_GRAPH_KERNEL_ADAPTER_EXPANDER_H_
#define MINDSPORE_CCSRC_COMMON_GRAPH_KERNEL_ADAPTER_EXPANDER_H_
#include <memory>
#include "common/graph_kernel/core/expander.h"
#include "backend/common/graph_kernel/core/expander.h"
#include "ir/func_graph.h"
#include "include/backend/visible.h"
#include <nlohmann/json.hpp>

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/fake_abstract_shape.h"
#include "backend/common/graph_kernel/adapter/fake_abstract_shape.h"
#include <sstream>
#include "include/common/utils/utils.h"

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/adapter/graph_kernel_expander_with_py.h"
#include "backend/common/graph_kernel/adapter/graph_kernel_expander_with_py.h"
#include <string>
#include <utility>
@ -23,8 +23,8 @@
#include <algorithm>
#include "utils/ms_context.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
namespace mindspore::graphkernel {
std::vector<PrimitivePtr> GraphKernelExpanderWithPy::GetExpanderOps() {
std::vector<OpWithLevel> expand_ops_with_level = {

View File

@ -19,8 +19,8 @@
#include <vector>
#include <string>
#include <nlohmann/json.hpp>
#include "common/graph_kernel/core/graph_kernel_expander.h"
#include "common/graph_kernel/adapter/expander.h"
#include "backend/common/graph_kernel/core/graph_kernel_expander.h"
#include "backend/common/graph_kernel/adapter/expander.h"
#include "ir/func_graph.h"
namespace mindspore::graphkernel {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/graph_kernel_optimization.h"
#include "backend/common/graph_kernel/adapter/graph_kernel_optimization.h"
#include <algorithm>
#include <vector>
@ -22,46 +22,46 @@
#include "ir/func_graph.h"
#include "utils/ms_context.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/add_atomic_clean.h"
#include "common/graph_kernel/add_stitch_atomic_clean_gpu.h"
#include "common/graph_kernel/core/arithmetic_simplify.h"
#include "common/graph_kernel/core/graph_kernel_cluster.h"
#include "common/graph_kernel/core/eliminate_redundant_output.h"
#include "common/graph_kernel/insert_pad.h"
#include "common/graph_kernel/adapter/graph_kernel_splitter_with_py.h"
#include "common/graph_kernel/adapter/graph_kernel_expander_with_py.h"
#include "common/graph_kernel/adapter/callback_impl.h"
#include "common/graph_kernel/cast_matmul_fusion.h"
#include "common/graph_kernel/raise_reduction_precision.h"
#include "common/graph_kernel/graph_kernel_cse.h"
#include "common/graph_kernel/core/shape_ops_splitter.h"
#include "common/graph_kernel/value_graph_binder.h"
#include "common/graph_kernel/parallel_fusion.h"
#include "common/graph_kernel/optimize_assign.h"
#include "common/graph_kernel/split_umonad.h"
#include "common/graph_kernel/reorder_ops.h"
#include "common/graph_kernel/core/update_state_formatter.h"
#include "common/graph_kernel/axis_normalizer.h"
#include "common/graph_kernel/decrease_compute_precision.h"
#include "common/graph_kernel/decrease_transfer_precision.h"
#include "common/graph_kernel/csr_atomic_add.h"
#include "common/graph_kernel/tsa_atomic_add_to_first_tensor.h"
#include "common/graph_kernel/uss_atomic_add.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/add_atomic_clean.h"
#include "backend/common/graph_kernel/add_stitch_atomic_clean_gpu.h"
#include "backend/common/graph_kernel/core/arithmetic_simplify.h"
#include "backend/common/graph_kernel/core/graph_kernel_cluster.h"
#include "backend/common/graph_kernel/core/eliminate_redundant_output.h"
#include "backend/common/graph_kernel/insert_pad.h"
#include "backend/common/graph_kernel/adapter/graph_kernel_splitter_with_py.h"
#include "backend/common/graph_kernel/adapter/graph_kernel_expander_with_py.h"
#include "backend/common/graph_kernel/adapter/callback_impl.h"
#include "backend/common/graph_kernel/cast_matmul_fusion.h"
#include "backend/common/graph_kernel/raise_reduction_precision.h"
#include "backend/common/graph_kernel/graph_kernel_cse.h"
#include "backend/common/graph_kernel/core/shape_ops_splitter.h"
#include "backend/common/graph_kernel/value_graph_binder.h"
#include "backend/common/graph_kernel/parallel_fusion.h"
#include "backend/common/graph_kernel/optimize_assign.h"
#include "backend/common/graph_kernel/split_umonad.h"
#include "backend/common/graph_kernel/reorder_ops.h"
#include "backend/common/graph_kernel/core/update_state_formatter.h"
#include "backend/common/graph_kernel/axis_normalizer.h"
#include "backend/common/graph_kernel/decrease_compute_precision.h"
#include "backend/common/graph_kernel/decrease_transfer_precision.h"
#include "backend/common/graph_kernel/csr_atomic_add.h"
#include "backend/common/graph_kernel/tsa_atomic_add_to_first_tensor.h"
#include "backend/common/graph_kernel/uss_atomic_add.h"
#include "backend/common/pass/getitem_tuple.h"
#include "common/graph_kernel/core/graph_kernel_pass_manager.h"
#include "common/graph_kernel/core/transform_op_optimizer.h"
#include "common/graph_kernel/rewrite_output_shape.h"
#include "common/graph_kernel/graph_kernel_recompute.h"
#include "common/graph_kernel/reduce_fake_out_mem.h"
#include "common/graph_kernel/depend_elimination.h"
#include "common/graph_kernel/tensor_inplace.h"
#include "common/graph_kernel/floatstatus_addn_fusion.h"
#include "common/graph_kernel/parallel_optimizer.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/compact_tensor_liveness.h"
#include "backend/common/graph_kernel/core/graph_kernel_pass_manager.h"
#include "backend/common/graph_kernel/core/transform_op_optimizer.h"
#include "backend/common/graph_kernel/rewrite_output_shape.h"
#include "backend/common/graph_kernel/graph_kernel_recompute.h"
#include "backend/common/graph_kernel/reduce_fake_out_mem.h"
#include "backend/common/graph_kernel/depend_elimination.h"
#include "backend/common/graph_kernel/tensor_inplace.h"
#include "backend/common/graph_kernel/floatstatus_addn_fusion.h"
#include "backend/common/graph_kernel/parallel_optimizer.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/compact_tensor_liveness.h"
#ifdef ENABLE_AKG
#include "common/graph_kernel/graph_kernel_build.h"
#include "backend/common/graph_kernel/graph_kernel_build.h"
#endif
namespace mindspore::graphkernel {
using opt::CommonSubexpressionElimination;

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/graph_kernel_splitter_with_py.h"
#include "backend/common/graph_kernel/adapter/graph_kernel_splitter_with_py.h"
#include <algorithm>
#include <vector>
@ -21,15 +21,16 @@
#include <memory>
#include <utility>
#include <map>
#include <set>
#include <nlohmann/json.hpp>
#include "utils/hash_map.h"
#include "utils/ms_context.h"
#include "include/common/utils/python_adapter.h"
#include "kernel/akg/akg_kernel_json_generator.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
namespace mindspore::graphkernel {
struct StitchInfo {

View File

@ -17,7 +17,7 @@
#define MINDSPORE_CCSRC_BACKEND_OPTIMIZER_GRAPH_KERNEL_ADAPTER_GRAPH_KERNEL_SPLITTER_WITH_PY_H_
#include <memory>
#include <string>
#include "common/graph_kernel/core/graph_kernel_splitter.h"
#include "backend/common/graph_kernel/core/graph_kernel_splitter.h"
namespace mindspore::graphkernel {
class GraphKernelSplitterWithPy : public GraphKernelSplitter {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/split_model_ascend.h"
#include "backend/common/graph_kernel/adapter/split_model_ascend.h"
#include <memory>
#include <string>
#include "utils/ms_context.h"

View File

@ -16,7 +16,7 @@
#ifndef MINDSPORE_CCSRC_COMMON_GRAPH_KERNEL_ADAPTER_SPLIT_MODEL_ASCEND_H_
#define MINDSPORE_CCSRC_COMMON_GRAPH_KERNEL_ADAPTER_SPLIT_MODEL_ASCEND_H_
#include "common/graph_kernel/split_model/split_model_factory.h"
#include "backend/common/graph_kernel/split_model/split_model_factory.h"
namespace mindspore::graphkernel::inner {
class SplitModelAscend : public SplitModel {
public:

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/adapter/split_model_cpu.h"
#include "backend/common/graph_kernel/adapter/split_model_cpu.h"
#include <memory>
#include "utils/ms_context.h"

View File

@ -16,7 +16,7 @@
#ifndef MINDSPORE_CCSRC_BACKEND_OPTIMIZER_GRAPH_KERNEL_ADAPTER_SPLIT_MODEL_CPU_H_
#define MINDSPORE_CCSRC_BACKEND_OPTIMIZER_GRAPH_KERNEL_ADAPTER_SPLIT_MODEL_CPU_H_
#include "common/graph_kernel/split_model/split_model_factory.h"
#include "backend/common/graph_kernel/split_model/split_model_factory.h"
namespace mindspore::graphkernel::inner {
class SplitModelCpu : public SplitModel {
public:

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/add_atomic_clean.h"
#include "backend/common/graph_kernel/add_atomic_clean.h"
#include <algorithm>
#include <functional>
@ -30,8 +30,8 @@
#include "include/common/utils/utils.h"
#include "utils/log_adapter.h"
#include "kernel/kernel.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/session/kernel_graph.h"
#include "include/common/debug/anf_ir_dump.h"
#include "kernel/common_utils.h"

View File

@ -23,8 +23,8 @@
#include <string>
#include "backend/common/optimizer/optimizer.h"
#include "backend/common/session/kernel_graph.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/inplace_assign_builder.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/inplace_assign_builder.h"
namespace mindspore::graphkernel {
class AtomicAddChecker {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/add_stitch_atomic_clean_gpu.h"
#include "backend/common/graph_kernel/add_stitch_atomic_clean_gpu.h"
#include <algorithm>
#include <string>
@ -23,8 +23,8 @@
#include "utils/log_adapter.h"
#include "kernel/kernel.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/session/kernel_graph.h"
namespace mindspore::graphkernel {

View File

@ -22,7 +22,7 @@
#include <utility>
#include <vector>
#include "backend/common/optimizer/optimizer.h"
#include "common/graph_kernel/add_atomic_clean.h"
#include "backend/common/graph_kernel/add_atomic_clean.h"
#include "backend/common/session/kernel_graph.h"
namespace mindspore::graphkernel {

View File

@ -13,12 +13,12 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/axis_normalizer.h"
#include "backend/common/graph_kernel/axis_normalizer.h"
#include <algorithm>
#include <vector>
#include "ir/scalar.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/session/anf_runtime_algorithm.h"
#include "include/common/utils/anfalgo.h"

View File

@ -13,14 +13,14 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/cast_matmul_fusion.h"
#include "backend/common/graph_kernel/cast_matmul_fusion.h"
#include <vector>
#include <string>
#include "backend/common/session/anf_runtime_algorithm.h"
#include "include/common/utils/anfalgo.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
namespace mindspore::graphkernel {
namespace {

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/compact_tensor_liveness.h"
#include "backend/common/graph_kernel/compact_tensor_liveness.h"
#include "include/common/utils/utils.h"
#include "include/common/utils/anfalgo.h"

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/arithmetic_simplify.h"
#include "backend/common/graph_kernel/core/arithmetic_simplify.h"
#include <algorithm>
#include <list>
@ -26,11 +26,11 @@
#include "utils/hash_map.h"
#include "utils/hash_set.h"
#include "utils/anf_utils.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/model/op_node.h"
#include "common/graph_kernel/model/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/model/op_node.h"
#include "backend/common/graph_kernel/model/graph_builder.h"
namespace mindspore::graphkernel {
// operator which follows commutative rules

View File

@ -23,7 +23,7 @@
#include "utils/hash_map.h"
#include "backend/common/optimizer/pass.h"
#include "ir/func_graph.h"
#include "common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
namespace mindspore::graphkernel {
class PatternTree;

View File

@ -13,8 +13,8 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "mindspore/ccsrc/common/graph_kernel/core/convert_op_input_attr.h"
#include "mindspore/ccsrc/common/graph_kernel/core/graph_kernel_callback.h"
#include "mindspore/ccsrc/backend/common/graph_kernel/core/convert_op_input_attr.h"
#include "mindspore/ccsrc/backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "mindspore/core/ops/core_ops.h"
#include "ops/primitive_c.h"

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/eliminate_redundant_output.h"
#include "backend/common/graph_kernel/core/eliminate_redundant_output.h"
#include <memory>
#include <algorithm>
@ -25,8 +25,8 @@
#include "ir/anf.h"
#include "ir/graph_utils.h"
#include "utils/anf_utils.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
namespace mindspore::graphkernel {
namespace {

View File

@ -14,14 +14,14 @@
* limitations under the License.
*/
#include "common/graph_kernel/core/expander.h"
#include "backend/common/graph_kernel/core/expander.h"
#include <string>
#include "utils/anf_utils.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/convert_op_input_attr.h"
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel {
ExpanderPtr WrapExpander(const ExpanderPtr &base, const ExpanderCreatorFuncList &deco_creators) {

View File

@ -19,7 +19,7 @@
#include <memory>
#include "ir/func_graph.h"
#include "include/backend/visible.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
namespace mindspore::graphkernel {
class BACKEND_EXPORT Expander {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
#include <algorithm>
#include <memory>
@ -27,8 +27,8 @@
#include "include/common/utils/utils.h"
#include "utils/anf_utils.h"
#include "utils/ordered_set.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "ir/func_graph_cloner.h"
namespace mindspore::graphkernel {

View File

@ -23,7 +23,7 @@
#include "ir/anf.h"
#include "ir/dtype/type_id.h"
#include "utils/shape_utils.h"
#include "common/graph_kernel/model/node.h"
#include "backend/common/graph_kernel/model/node.h"
#include "include/backend/visible.h"
namespace mindspore::graphkernel {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_kernel_cluster.h"
#include "backend/common/graph_kernel/core/graph_kernel_cluster.h"
#include <algorithm>
#include <set>
@ -28,11 +28,11 @@
#include "utils/anf_utils.h"
#include "utils/ms_context.h"
#include "utils/file_utils.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/convert_op_input_attr.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
namespace mindspore::graphkernel {
std::vector<PrimitivePtr> GraphKernelCluster::GetClusterOps() {

View File

@ -14,16 +14,16 @@
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_kernel_expander.h"
#include "backend/common/graph_kernel/core/graph_kernel_expander.h"
#include <string>
#include <vector>
#include <algorithm>
#include "utils/anf_utils.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/convert_op_input_attr.h"
namespace mindspore::graphkernel {
AnfNodePtr GraphKernelExpander::CreateExpandedNode(const CNodePtr &node, const std::string &name) const {

View File

@ -20,8 +20,8 @@
#include <string>
#include "backend/common/optimizer/pass.h"
#include "ir/func_graph.h"
#include "common/graph_kernel/core/expander.h"
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/core/expander.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel {
class GraphKernelExpander : public opt::Pass {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_kernel_pass_manager.h"
#include "backend/common/graph_kernel/core/graph_kernel_pass_manager.h"
#include <iomanip>
#include <algorithm>

View File

@ -21,7 +21,7 @@
#include <string>
#include <memory>
#include "common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/optimizer/pass_manager.h"
namespace mindspore::graphkernel {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_kernel_splitter.h"
#include "backend/common/graph_kernel/core/graph_kernel_splitter.h"
#include <algorithm>
#include <vector>
#include <string>
@ -23,10 +23,10 @@
#include "utils/anf_utils.h"
#include "utils/hash_map.h"
#include "utils/hash_set.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/convert_op_input_attr.h"
#include "common/graph_kernel/split_model/split_model_factory.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/convert_op_input_attr.h"
#include "backend/common/graph_kernel/split_model/split_model_factory.h"
namespace mindspore::graphkernel {
namespace {

View File

@ -19,7 +19,7 @@
#include <string>
#include "ir/func_graph.h"
#include "backend/common/optimizer/pass.h"
#include "common/graph_kernel/core/split_schemer.h"
#include "backend/common/graph_kernel/core/split_schemer.h"
namespace mindspore::graphkernel {
class GraphKernelSplitter : public opt::Pass {

View File

@ -13,17 +13,18 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include <map>
#include <memory>
#include <sstream>
#include <utility>
#include <unordered_map>
#include <algorithm>
#include "mindspore/core/ops/core_ops.h"
#include "utils/anf_utils.h"
#include "utils/ms_context.h"
#include "common/graph_kernel/model/op_node.h"
#include "common/graph_kernel/model/graph_builder.h"
#include "backend/common/graph_kernel/model/op_node.h"
#include "backend/common/graph_kernel/model/graph_builder.h"
#include "runtime/hardware/device_context_manager.h"
namespace mindspore::graphkernel {

View File

@ -24,8 +24,8 @@
#include "ir/anf.h"
#include "ir/func_graph.h"
#include "utils/hash_map.h"
#include "common/graph_kernel/model/lite_graph.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
namespace mindspore::prim {
GVAR_DEF(PrimitivePtr, kPrimElemAny, std::make_shared<Primitive>("ElemAny"));

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/shape_ops_splitter.h"
#include "backend/common/graph_kernel/core/shape_ops_splitter.h"
#include <algorithm>
#include <vector>

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/split_schemer.h"
#include "backend/common/graph_kernel/core/split_schemer.h"
#include "ops/core_ops.h"
namespace mindspore::graphkernel {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/transform_op_optimizer.h"
#include "backend/common/graph_kernel/core/transform_op_optimizer.h"
#include <algorithm>
#include <vector>
#include <queue>
@ -25,10 +25,10 @@
#include "mindspore/core/ops/core_ops.h"
#include "ir/graph_utils.h"
#include "utils/anf_utils.h"
#include "common/graph_kernel/model/lite_graph.h"
#include "common/graph_kernel/model/op_register.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/model/op_register.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
namespace mindspore::graphkernel {
namespace {

View File

@ -22,7 +22,7 @@
#include <functional>
#include "backend/common/optimizer/pass.h"
#include "ir/func_graph.h"
#include "common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
namespace mindspore::graphkernel {
using inner::NodePtr;

View File

@ -13,11 +13,11 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/tuning_splitter.h"
#include "backend/common/graph_kernel/core/tuning_splitter.h"
#include <fstream>
#include <vector>
#include "utils/ms_utils.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
namespace mindspore::graphkernel {
bool TuningSplitSchemer::ReadCache(const std::string &filename, nlohmann::json *result) const {

View File

@ -19,7 +19,7 @@
#include <string>
#include <nlohmann/json.hpp>
#include "ir/func_graph.h"
#include "common/graph_kernel/core/split_schemer.h"
#include "backend/common/graph_kernel/core/split_schemer.h"
namespace mindspore::graphkernel {
class TuningSplitSchemer : public CommonSplitSchemer {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/core/update_state_formatter.h"
#include "backend/common/graph_kernel/core/update_state_formatter.h"
#include <vector>
#include <set>
@ -21,9 +21,9 @@
#include <utility>
#include <algorithm>
#include "ir/anf.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "common/graph_kernel/core/graph_kernel_callback.h"
#include "common/graph_kernel/core/eliminate_redundant_output.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/core/graph_kernel_callback.h"
#include "backend/common/graph_kernel/core/eliminate_redundant_output.h"
namespace mindspore::graphkernel {
AnfNodePtrList GetUpdateStateList(const FuncGraphPtr &func_graph) {

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/csr_atomic_add.h"
#include "backend/common/graph_kernel/csr_atomic_add.h"
#include <memory>
#include <vector>
#include "mindspore/core/ops/core_ops.h"
@ -22,7 +22,7 @@
#include "include/common/utils/utils.h"
#include "kernel/kernel.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/session/kernel_graph.h"
namespace mindspore::graphkernel {

View File

@ -20,7 +20,7 @@
#include <memory>
#include <tuple>
#include "backend/common/optimizer/optimizer.h"
#include "common/graph_kernel/add_atomic_clean.h"
#include "backend/common/graph_kernel/add_atomic_clean.h"
#include "backend/common/session/kernel_graph.h"
namespace mindspore::graphkernel {

View File

@ -22,10 +22,10 @@
#include "backend/common/optimizer/helper.h"
#include "backend/common/session/anf_runtime_algorithm.h"
#include "include/common/utils/anfalgo.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/decrease_compute_precision.h"
#include "backend/common/graph_kernel/decrease_compute_precision.h"
namespace mindspore::graphkernel {
// Add CastCNode

View File

@ -21,13 +21,13 @@
#include <utility>
#include "mindspore/core/ops/core_ops.h"
#include "include/common/utils/utils.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "ir/tensor.h"
#include "ir/manager.h"
#include "kernel/kernel_build_info.h"
#include "kernel/common_utils.h"
#include "runtime/device/kernel_info.h"
#include "common/graph_kernel/decrease_transfer_precision.h"
#include "backend/common/graph_kernel/decrease_transfer_precision.h"
namespace mindspore::graphkernel {
static const size_t GK_MIN_SIZE = 2; // 2

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/depend_elimination.h"
#include "backend/common/graph_kernel/depend_elimination.h"
#include "include/common/utils/utils.h"
namespace mindspore::graphkernel {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -18,8 +18,8 @@
#include <vector>
#include <memory>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/utils.h"
namespace mindspore::graphkernel::expanders {
class BiasAdd : public OpDesc {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -18,9 +18,9 @@
#include <vector>
#include <memory>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "common/graph_kernel/expanders/utils.h"
#include "common/graph_kernel/expanders/custom_op_utils.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/custom_op_utils.h"
namespace mindspore::graphkernel::expanders {
class LU : public OpDesc {

View File

@ -21,7 +21,7 @@
#include <memory>
#include "utils/hash_map.h"
#include "common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/utils.h"
#include "include/backend/visible.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -16,7 +16,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel::expanders {
class ExpandDims : public OpDesc {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel::expanders {
class SiLU : public OpDesc {

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel::expanders {
class SiLUGrad : public OpDesc {

View File

@ -16,8 +16,8 @@
#include <memory>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/utils.h"
#include "kernel/common_utils.h"
namespace mindspore::graphkernel::expanders {

View File

@ -18,7 +18,7 @@
#include <vector>
#include <limits>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -18,9 +18,9 @@
#include <vector>
#include <memory>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "common/graph_kernel/expanders/utils.h"
#include "common/graph_kernel/expanders/custom_op_utils.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/custom_op_utils.h"
namespace mindspore::graphkernel::expanders {
class SolveTriangular : public OpDesc {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
namespace mindspore::graphkernel::expanders {
class StandardNormal : public OpDesc {

View File

@ -13,15 +13,15 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/expanders/utils.h"
#include "backend/common/graph_kernel/expanders/utils.h"
#include <algorithm>
#include <string>
#include <vector>
#include <unordered_set>
#include "common/graph_kernel/model/lite_graph.h"
#include "common/graph_kernel/model/node.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/model/node.h"
namespace mindspore::graphkernel::expanders {
constexpr int OFFSET1 = 1;

View File

@ -21,9 +21,9 @@
#include <vector>
#include <utility>
#include "common/graph_kernel/model/lite_graph.h"
#include "common/graph_kernel/model/node.h"
#include "common/graph_kernel/model/graph_builder.h"
#include "backend/common/graph_kernel/model/lite_graph.h"
#include "backend/common/graph_kernel/model/node.h"
#include "backend/common/graph_kernel/model/graph_builder.h"
namespace mindspore::graphkernel::expanders {
using inner::NodePtr;

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -17,7 +17,7 @@
#include <memory>
#include <vector>
#include "common/graph_kernel/expanders/op_desc_registry.h"
#include "backend/common/graph_kernel/expanders/op_desc_registry.h"
#include "ir/dtype.h"
namespace mindspore::graphkernel::expanders {

View File

@ -13,7 +13,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "common/graph_kernel/floatstatus_addn_fusion.h"
#include "backend/common/graph_kernel/floatstatus_addn_fusion.h"
#include <vector>
#include <string>
@ -23,11 +23,11 @@
#include "backend/common/session/anf_runtime_algorithm.h"
#include "include/common/utils/anfalgo.h"
#include "kernel/common_utils.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/adapter/expander.h"
#include "common/graph_kernel/core/graph_builder.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/adapter/expander.h"
#include "backend/common/graph_kernel/core/graph_builder.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
namespace mindspore::graphkernel {
namespace {

View File

@ -18,7 +18,7 @@
#include <string>
#include "backend/common/optimizer/pass.h"
#include "common/graph_kernel/inplace_assign_builder.h"
#include "backend/common/graph_kernel/inplace_assign_builder.h"
namespace mindspore::graphkernel {
/**

View File

@ -14,7 +14,7 @@
* limitations under the License.
*/
#ifdef ENABLE_AKG
#include "common/graph_kernel/graph_kernel_build.h"
#include "backend/common/graph_kernel/graph_kernel_build.h"
#include <fstream>
#include <utility>
@ -24,9 +24,9 @@
#include "include/common/utils/anfalgo.h"
#include "backend/common/session/anf_runtime_algorithm.h"
#include "kernel/akg/akg_kernel_json_generator.h"
#include "common/graph_kernel/graph_kernel_helper.h"
#include "common/graph_kernel/graph_kernel_flags.h"
#include "common/graph_kernel/core/graph_kernel_utils.h"
#include "backend/common/graph_kernel/graph_kernel_helper.h"
#include "backend/common/graph_kernel/graph_kernel_flags.h"
#include "backend/common/graph_kernel/core/graph_kernel_utils.h"
#include "kernel/akg/akg_kernel_build_manager.h"
namespace mindspore::graphkernel {

Some files were not shown because too many files have changed in this diff Show More