forked from mindspore-Ecosystem/mindspore
move graph_kernel & mem_reuse to backend
Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
parent
19735e8afd
commit
a1a3f459f9
|
@ -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
|
||||
|
|
|
@ -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})
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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"
|
|
@ -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>
|
|
@ -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"
|
||||
|
|
@ -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 = {
|
|
@ -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 {
|
|
@ -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;
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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"
|
|
@ -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:
|
|
@ -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"
|
||||
|
|
@ -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:
|
|
@ -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"
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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"
|
||||
|
|
@ -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 {
|
|
@ -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"
|
||||
|
|
@ -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
|
|
@ -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;
|
|
@ -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"
|
|
@ -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 {
|
|
@ -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) {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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() {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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>
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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"));
|
|
@ -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>
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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;
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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) {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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
|
|
@ -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
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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;
|
|
@ -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;
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
|
@ -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 {
|
||||
/**
|
|
@ -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
Loading…
Reference in New Issue