forked from mindspore-Ecosystem/mindspore
!3053 mindspore path adjust
Merge pull request !3053 from liubuyu/master
This commit is contained in:
commit
c99cc0dfa1
|
@ -1,4 +1,5 @@
|
||||||
## common setting
|
## common setting
|
||||||
|
include_directories(${CMAKE_SOURCE_DIR}/mindspore/core)
|
||||||
include_directories(${CMAKE_CURRENT_SOURCE_DIR})
|
include_directories(${CMAKE_CURRENT_SOURCE_DIR})
|
||||||
include_directories(${CMAKE_BINARY_DIR})
|
include_directories(${CMAKE_BINARY_DIR})
|
||||||
link_directories(${CMAKE_SOURCE_DIR}/build/mindspore/graphengine)
|
link_directories(${CMAKE_SOURCE_DIR}/build/mindspore/graphengine)
|
||||||
|
@ -35,20 +36,20 @@ if(ENABLE_GPU)
|
||||||
include_directories(${CUDNN_PATH} ${CUDA_PATH} ${CUDA_INCLUDE_DIRS})
|
include_directories(${CUDNN_PATH} ${CUDA_PATH} ${CUDA_INCLUDE_DIRS})
|
||||||
|
|
||||||
file(GLOB_RECURSE GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
file(GLOB_RECURSE GPU_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
|
||||||
"device/gpu/*.cc"
|
"runtime/device/gpu/*.cc"
|
||||||
"device/gpu/*.cu"
|
"runtime/device/gpu/*.cu"
|
||||||
"kernel/gpu/*.cu"
|
"backend/kernel_compiler/gpu/*.cu"
|
||||||
"kernel/akg/gpu/*.cc"
|
"backend/kernel_compiler/akg/gpu/*.cc"
|
||||||
"kernel/akg/akg_kernel_build.cc"
|
"backend/kernel_compiler/akg/akg_kernel_build.cc"
|
||||||
"kernel/akg/akg_kernel_attrs_process.cc"
|
"backend/kernel_compiler/akg/akg_kernel_attrs_process.cc"
|
||||||
)
|
)
|
||||||
|
|
||||||
list(APPEND CUDA_NVCC_FLAGS -arch=sm_53)
|
list(APPEND CUDA_NVCC_FLAGS -arch=sm_53)
|
||||||
list(REMOVE_ITEM GPU_SRC_LIST "device/gpu/blocking_queue.cc" "device/gpu/gpu_buffer_mgr.cc")
|
list(REMOVE_ITEM GPU_SRC_LIST "runtime/device/gpu/blocking_queue.cc" "runtime/device/gpu/gpu_buffer_mgr.cc")
|
||||||
list(REMOVE_ITEM GPU_SRC_LIST "device/gpu/mpi/mpi_initializer.cc"
|
list(REMOVE_ITEM GPU_SRC_LIST "runtime/device/gpu/mpi/mpi_initializer.cc"
|
||||||
"device/gpu/distribution/collective_wrapper.cc"
|
"runtime/device/gpu/distribution/collective_wrapper.cc"
|
||||||
"device/gpu/distribution/mpi_wrapper.cc"
|
"runtime/device/gpu/distribution/mpi_wrapper.cc"
|
||||||
"device/gpu/distribution/nccl_wrapper.cc"
|
"runtime/device/gpu/distribution/nccl_wrapper.cc"
|
||||||
)
|
)
|
||||||
|
|
||||||
set(NVCC_TMP_CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})
|
set(NVCC_TMP_CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS})
|
||||||
|
@ -101,15 +102,15 @@ if (ENABLE_DUMP_PROTO)
|
||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
if (ENABLE_D)
|
if (ENABLE_D)
|
||||||
include_directories("${CMAKE_BINARY_DIR}/kernel/aicpu")
|
include_directories("${CMAKE_BINARY_DIR}/backend/kernel_compiler/aicpu")
|
||||||
include_directories("${CMAKE_BINARY_DIR}/predict/generator/ir")
|
include_directories("${CMAKE_BINARY_DIR}/predict/generator/ir")
|
||||||
file(GLOB_RECURSE PROTO_IN RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "kernel/aicpu/proto/*.proto")
|
file(GLOB_RECURSE PROTO_IN RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "backend/kernel_compiler/aicpu/proto/*.proto")
|
||||||
ms_protobuf_generate(PROTOSRCS PROTOHDRS ${PROTO_IN})
|
ms_protobuf_generate(PROTOSRCS PROTOHDRS ${PROTO_IN})
|
||||||
|
|
||||||
file(GLOB_RECURSE PROTO_INNER RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "predict/proto/*.proto")
|
file(GLOB_RECURSE PROTO_INNER RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "predict/proto/*.proto")
|
||||||
ms_protobuf_generate(PREDICT_PROTOSRCS PREDICT_PROTOHDRS ${PROTO_INNER})
|
ms_protobuf_generate(PREDICT_PROTOSRCS PREDICT_PROTOHDRS ${PROTO_INNER})
|
||||||
|
|
||||||
file(GLOB_RECURSE PROTO_DUMP RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "device/ascend/dump/proto/*.proto")
|
file(GLOB_RECURSE PROTO_DUMP RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "runtime/device/ascend/dump/proto/*.proto")
|
||||||
ms_protobuf_generate(DUMP_PROTOSRCS PROTOHDRS ${PROTO_DUMP})
|
ms_protobuf_generate(DUMP_PROTOSRCS PROTOHDRS ${PROTO_DUMP})
|
||||||
|
|
||||||
list(APPEND MINDSPORE_PROTO_LIST ${PROTOSRCS})
|
list(APPEND MINDSPORE_PROTO_LIST ${PROTOSRCS})
|
||||||
|
@ -125,18 +126,32 @@ if (MINDSPORE_PROTO_LIST)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
## make sub objects
|
## make sub objects
|
||||||
set(SUB_COMP
|
set(SUB_COMP
|
||||||
transform pre_activate parallel pipeline device kernel common debug gvar ir onnx operator optimizer predict
|
transform/graph_ir
|
||||||
pybind_api pynative session utils vm base abstract
|
transform/onnx
|
||||||
|
backend/optimizer
|
||||||
|
backend/kernel_compiler
|
||||||
|
backend/session
|
||||||
|
runtime/device
|
||||||
|
frontend/optimizer
|
||||||
|
frontend/parallel
|
||||||
|
frontend/operator
|
||||||
|
pipeline/jit
|
||||||
|
pipeline/pynative
|
||||||
|
common debug gvar predict pybind_api utils vm base abstract
|
||||||
)
|
)
|
||||||
|
|
||||||
foreach (_comp ${SUB_COMP})
|
foreach (_comp ${SUB_COMP})
|
||||||
add_subdirectory(${_comp})
|
add_subdirectory(${_comp})
|
||||||
if (TARGET _mindspore_${_comp}_obj)
|
string(REPLACE "/" "_" sub ${_comp})
|
||||||
list(APPEND SUB_OBJECTS_SRC $<TARGET_OBJECTS:_mindspore_${_comp}_obj>)
|
if (TARGET _mindspore_${sub}_obj)
|
||||||
add_dependencies(_mindspore_${_comp}_obj proto_input flat_input)
|
list(APPEND SUB_OBJECTS_SRC $<TARGET_OBJECTS:_mindspore_${sub}_obj>)
|
||||||
|
add_dependencies(_mindspore_${sub}_obj proto_input flat_input)
|
||||||
endif ()
|
endif ()
|
||||||
endforeach ()
|
endforeach ()
|
||||||
|
add_subdirectory(${CMAKE_SOURCE_DIR}/mindspore/core/ir ir)
|
||||||
|
list(APPEND SUB_OBJECTS_SRC $<TARGET_OBJECTS:_mindspore_ir_obj>)
|
||||||
|
add_dependencies(_mindspore_ir_obj proto_input flat_input)
|
||||||
|
|
||||||
set_property(SOURCE ${SUB_OBJECTS_SRC} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_ME)
|
set_property(SOURCE ${SUB_OBJECTS_SRC} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_ME)
|
||||||
add_library(mindspore STATIC ${SUB_OBJECTS_SRC})
|
add_library(mindspore STATIC ${SUB_OBJECTS_SRC})
|
||||||
|
@ -207,8 +222,8 @@ endif()
|
||||||
|
|
||||||
# set c_expression building
|
# set c_expression building
|
||||||
set(CMAKE_BUILD_WITH_INSTALL_RPATH TRUE)
|
set(CMAKE_BUILD_WITH_INSTALL_RPATH TRUE)
|
||||||
set_property(SOURCE "pipeline/init.cc" PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_PIPELINE)
|
set_property(SOURCE "pipeline/jit/init.cc" PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_PIPELINE)
|
||||||
pybind11_add_module(_c_expression "pipeline/init.cc")
|
pybind11_add_module(_c_expression "pipeline/jit/init.cc")
|
||||||
|
|
||||||
MESSAGE(STATUS "operation system is ${CMAKE_SYSTEM}")
|
MESSAGE(STATUS "operation system is ${CMAKE_SYSTEM}")
|
||||||
if (CMAKE_SYSTEM_NAME MATCHES "Linux")
|
if (CMAKE_SYSTEM_NAME MATCHES "Linux")
|
||||||
|
@ -265,8 +280,8 @@ if (ENABLE_CPU)
|
||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
if (ENABLE_MINDDATA)
|
if (ENABLE_MINDDATA)
|
||||||
add_subdirectory(mindrecord)
|
add_subdirectory(minddata/mindrecord)
|
||||||
add_subdirectory(dataset)
|
add_subdirectory(minddata/dataset)
|
||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
# build inference
|
# build inference
|
||||||
|
@ -275,7 +290,7 @@ set(LOAD_ONNX_SRC
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/utils/load_onnx/anf_model_parser.cc
|
${CMAKE_CURRENT_SOURCE_DIR}/utils/load_onnx/anf_model_parser.cc
|
||||||
)
|
)
|
||||||
add_library(inference SHARED
|
add_library(inference SHARED
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/session/session.cc
|
${CMAKE_CURRENT_SOURCE_DIR}/backend/session/session.cc
|
||||||
${LOAD_ONNX_SRC}
|
${LOAD_ONNX_SRC}
|
||||||
)
|
)
|
||||||
target_link_libraries(inference PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}
|
target_link_libraries(inference PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}
|
||||||
|
|
|
@ -63,4 +63,4 @@ endif()
|
||||||
|
|
||||||
set_property(SOURCE ${KERNEL_SRC_LIST} ${CPU_SRC_LIST} ${GPU_SRC_LIST} ${D_SRC_LIST}
|
set_property(SOURCE ${KERNEL_SRC_LIST} ${CPU_SRC_LIST} ${GPU_SRC_LIST} ${D_SRC_LIST}
|
||||||
PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_KERNEL)
|
PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_KERNEL)
|
||||||
add_library(_mindspore_kernel_obj OBJECT ${KERNEL_SRC_LIST} ${CPU_SRC_LIST} ${GPU_SRC_LIST} ${D_SRC_LIST})
|
add_library(_mindspore_backend_kernel_compiler_obj OBJECT ${KERNEL_SRC_LIST} ${CPU_SRC_LIST} ${GPU_SRC_LIST} ${D_SRC_LIST})
|
|
@ -13,7 +13,7 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/aicpu/aicpu_kernel_build.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_kernel_build.h"
|
||||||
#include <google/protobuf/text_format.h>
|
#include <google/protobuf/text_format.h>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
@ -22,18 +22,18 @@
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <map>
|
#include <map>
|
||||||
#include "device/kernel_runtime.h"
|
#include "runtime/device/kernel_runtime.h"
|
||||||
#include "kernel/aicpu/aicpu_kernel_mod.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_kernel_mod.h"
|
||||||
#include "kernel/akg/akg_kernel_build.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_build.h"
|
||||||
#include "proto/tensor.pb.h"
|
#include "proto/tensor.pb.h"
|
||||||
#include "proto/tensor_shape.pb.h"
|
#include "proto/tensor_shape.pb.h"
|
||||||
#include "proto/attr.pb.h"
|
#include "proto/attr.pb.h"
|
||||||
#include "proto/node_def.pb.h"
|
#include "proto/node_def.pb.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/aicpu/aicpu_util.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_util.h"
|
||||||
#include "session/kernel_graph.h"
|
#include "backend/session/kernel_graph.h"
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -16,7 +16,7 @@
|
||||||
#ifndef MINDSPORE_MINDSPORE_CCSRC_KERNEL_AICPU_AICPU_KERNEL_BUILD_H_
|
#ifndef MINDSPORE_MINDSPORE_CCSRC_KERNEL_AICPU_AICPU_KERNEL_BUILD_H_
|
||||||
#define MINDSPORE_MINDSPORE_CCSRC_KERNEL_AICPU_AICPU_KERNEL_BUILD_H_
|
#define MINDSPORE_MINDSPORE_CCSRC_KERNEL_AICPU_AICPU_KERNEL_BUILD_H_
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,13 +14,13 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/aicpu/aicpu_kernel_metadata.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_kernel_metadata.h"
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "kernel/oplib/oplib.h"
|
#include "backend/kernel_compiler/oplib/oplib.h"
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
#include "kernel/aicpu/aicpu_util.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_util.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -20,7 +20,7 @@
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/kernel_build_info.h"
|
#include "backend/kernel_compiler/kernel_build_info.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/aicpu/aicpu_kernel_mod.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_kernel_mod.h"
|
||||||
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
@ -23,9 +23,9 @@
|
||||||
|
|
||||||
#include "runtime/mem.h"
|
#include "runtime/mem.h"
|
||||||
#include "runtime/rt.h"
|
#include "runtime/rt.h"
|
||||||
#include "kernel/aicpu/aicpu_kernel_build.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_kernel_build.h"
|
||||||
#include "utils/convert_utils.h"
|
#include "utils/convert_utils.h"
|
||||||
#include "kernel/aicpu/aicpu_util.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_util.h"
|
||||||
#include "utils/context/ms_context.h"
|
#include "utils/context/ms_context.h"
|
||||||
|
|
||||||
using AicpuTaskInfoPtr = std::shared_ptr<ge::model_runner::AicpuTaskInfo>;
|
using AicpuTaskInfoPtr = std::shared_ptr<ge::model_runner::AicpuTaskInfo>;
|
|
@ -18,8 +18,8 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "kernel/ascend_kernel_mod.h"
|
#include "backend/kernel_compiler/ascend_kernel_mod.h"
|
||||||
#include "kernel/aicpu/aicpu_util.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_util.h"
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
||||||
class AicpuOpKernelMod : public AscendKernelMod {
|
class AicpuOpKernelMod : public AscendKernelMod {
|
|
@ -13,14 +13,14 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/aicpu/aicpu_util.h"
|
#include "backend/kernel_compiler/aicpu/aicpu_util.h"
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "proto/types.pb.h"
|
#include "proto/types.pb.h"
|
||||||
#include "runtime/mem.h"
|
#include "runtime/mem.h"
|
||||||
#include "runtime/rt.h"
|
#include "runtime/rt.h"
|
||||||
#include "utils/convert_utils.h"
|
#include "utils/convert_utils.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -20,7 +20,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <map>
|
#include <map>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,11 +13,11 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/akg/akg_kernel_attrs_process.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_attrs_process.h"
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
#include "pre_activate/common/helper.h"
|
#include "backend/optimizer/common/helper.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -22,7 +22,7 @@
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
#include "ir/anf.h"
|
#include "ir/anf.h"
|
||||||
#include "utils/utils.h"
|
#include "utils/utils.h"
|
||||||
#include "operator/ops.h"
|
#include "frontend/operator/ops.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/akg_kernel_build.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_build.h"
|
||||||
#include <Python.h>
|
#include <Python.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
|
@ -35,8 +35,8 @@
|
||||||
#include "utils/convert_utils.h"
|
#include "utils/convert_utils.h"
|
||||||
#include "utils/any.h"
|
#include "utils/any.h"
|
||||||
#include "utils/utils.h"
|
#include "utils/utils.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
#include "kernel/akg/akg_kernel_attrs_process.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_attrs_process.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -22,11 +22,11 @@
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <map>
|
#include <map>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "ir/dtype.h"
|
#include "ir/dtype.h"
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
#include "kernel/oplib/oplib.h"
|
#include "backend/kernel_compiler/oplib/oplib.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,11 +14,11 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/akg_kernel_metadata.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_metadata.h"
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
#include "kernel/oplib/oplib.h"
|
#include "backend/kernel_compiler/oplib/oplib.h"
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -21,7 +21,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/kernel_build_info.h"
|
#include "backend/kernel_compiler/kernel_build_info.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/ascend/akg_ascend_kernel_build.h"
|
#include "backend/kernel_compiler/akg/ascend/akg_ascend_kernel_build.h"
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
@ -26,12 +26,12 @@
|
||||||
#include <Python.h>
|
#include <Python.h>
|
||||||
#include "ir/dtype.h"
|
#include "ir/dtype.h"
|
||||||
#include "ir/func_graph.h"
|
#include "ir/func_graph.h"
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
#include "kernel/tbe/tbe_utils.h"
|
#include "backend/kernel_compiler/tbe/tbe_utils.h"
|
||||||
#include "kernel/akg/ascend/akg_ascend_kernel_mod.h"
|
#include "backend/kernel_compiler/akg/ascend/akg_ascend_kernel_mod.h"
|
||||||
#include "kernel/akg/akg_kernel_attrs_process.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_attrs_process.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -22,8 +22,8 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <map>
|
#include <map>
|
||||||
#include "ir/anf.h"
|
#include "ir/anf.h"
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "kernel/akg/akg_kernel_build.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_build.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/ascend/akg_ascend_kernel_mod.h"
|
#include "backend/kernel_compiler/akg/ascend/akg_ascend_kernel_mod.h"
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <map>
|
#include <map>
|
|
@ -19,8 +19,8 @@
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/ascend_kernel_mod.h"
|
#include "backend/kernel_compiler/ascend_kernel_mod.h"
|
||||||
#include "kernel/tbe/tbe_utils.h"
|
#include "backend/kernel_compiler/tbe/tbe_utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,12 +14,12 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/gpu/akg_gpu_kernel_build.h"
|
#include "backend/kernel_compiler/akg/gpu/akg_gpu_kernel_build.h"
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "kernel/akg/akg_kernel_build.h"
|
#include "backend/kernel_compiler/akg/akg_kernel_build.h"
|
||||||
#include "kernel/akg/gpu/akg_gpu_kernel_mod.h"
|
#include "backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -16,7 +16,7 @@
|
||||||
|
|
||||||
#ifndef MINDSPORE_CCSRC_KERNEL_AKG_GPU_AKG_GPU_KERNEL_BUILD_H_
|
#ifndef MINDSPORE_CCSRC_KERNEL_AKG_GPU_AKG_GPU_KERNEL_BUILD_H_
|
||||||
#define MINDSPORE_CCSRC_KERNEL_AKG_GPU_AKG_GPU_KERNEL_BUILD_H_
|
#define MINDSPORE_CCSRC_KERNEL_AKG_GPU_AKG_GPU_KERNEL_BUILD_H_
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "base/base.h"
|
#include "base/base.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/akg/gpu/akg_gpu_kernel_mod.h"
|
#include "backend/kernel_compiler/akg/gpu/akg_gpu_kernel_mod.h"
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include "nlohmann/json.hpp"
|
#include "nlohmann/json.hpp"
|
|
@ -21,7 +21,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -20,7 +20,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "framework/ge_runtime/task_info.h"
|
#include "framework/ge_runtime/task_info.h"
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#ifdef ENABLE_DATA_DUMP
|
#ifdef ENABLE_DATA_DUMP
|
||||||
#include "debug/data_dump_parser.h"
|
#include "debug/data_dump_parser.h"
|
||||||
#endif
|
#endif
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
#include <map>
|
#include <map>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
@ -22,12 +22,12 @@
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include "nlohmann/json.hpp"
|
#include "nlohmann/json.hpp"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "ir/manager.h"
|
#include "ir/manager.h"
|
||||||
#include "ir/meta_tensor.h"
|
#include "ir/meta_tensor.h"
|
||||||
#include "ir/func_graph.h"
|
#include "ir/func_graph.h"
|
||||||
#include "operator/ops.h"
|
#include "frontend/operator/ops.h"
|
||||||
#include "utils/graph_utils.h"
|
#include "utils/graph_utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -26,9 +26,9 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "kernel/oplib/opinfo.h"
|
#include "backend/kernel_compiler/oplib/opinfo.h"
|
||||||
#include "kernel/kernel_build_info.h"
|
#include "backend/kernel_compiler/kernel_build_info.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,8 +14,8 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/addn_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/addn_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,8 +18,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_ADDN_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_ADDN_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/allgather_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/allgather_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "device/cpu/mpi/mpi_adapter.h"
|
#include "runtime/device/cpu/mpi/mpi_adapter.h"
|
||||||
#include "utils/log_adapter.h"
|
#include "utils/log_adapter.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_REDUCE_SCATTER_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_REDUCE_SCATTER_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/apply_momentum_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/apply_momentum_cpu_kernel.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,8 +13,8 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/argmax_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/argmax_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_ARGMAX_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_ARGMAX_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/bias_add_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/bias_add_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,8 +18,8 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,7 +14,7 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/bias_add_grad_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/bias_add_grad_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -19,8 +19,8 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,8 +14,8 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/concat_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/concat_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_CONCAT_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_CONCAT_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,7 +13,7 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -21,9 +21,9 @@
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include "kernel/kernel.h"
|
#include "backend/kernel_compiler/kernel.h"
|
||||||
#include "ir/anf.h"
|
#include "ir/anf.h"
|
||||||
#include "session/anf_runtime_algorithm.h"
|
#include "backend/session/anf_runtime_algorithm.h"
|
||||||
|
|
||||||
using mindspore::kernel::Address;
|
using mindspore::kernel::Address;
|
||||||
using mindspore::kernel::AddressPtr;
|
using mindspore::kernel::AddressPtr;
|
|
@ -14,13 +14,13 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
#include "device/kernel_info.h"
|
#include "runtime/device/kernel_info.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -24,8 +24,8 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "device/cpu/kernel_select_cpu.h"
|
#include "runtime/device/cpu/kernel_select_cpu.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,8 +13,8 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/debug_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/debug_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#ifdef ENABLE_DEBUGGER
|
#ifdef ENABLE_DEBUGGER
|
||||||
#include "debug/debugger/debugger.h"
|
#include "debug/debugger/debugger.h"
|
|
@ -18,8 +18,8 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,9 +14,9 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include "kernel/cpu/embedding_look_up_comm_grad_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/embedding_look_up_comm_grad_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "device/cpu/mpi/mpi_adapter.h"
|
#include "runtime/device/cpu/mpi/mpi_adapter.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_COMM_GRAD_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_COMM_GRAD_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -15,9 +15,9 @@
|
||||||
*/
|
*/
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "kernel/cpu/embedding_look_up_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/embedding_look_up_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "device/cpu/mpi/mpi_adapter.h"
|
#include "runtime/device/cpu/mpi/mpi_adapter.h"
|
||||||
#include "ir/primitive.h"
|
#include "ir/primitive.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,8 +13,8 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/equal_count_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/equal_count_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_EQUAL_COUNT_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_EQUAL_COUNT_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,8 +13,8 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/gather_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/gather_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_GATHER_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_GATHER_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,11 +13,11 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/conv2d_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,11 +13,11 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/conv2d_grad_filter_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_filter_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,10 +13,10 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/conv2d_grad_input_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_input_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,11 +13,11 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/lstm_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/lstm_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -24,7 +24,7 @@
|
||||||
#endif
|
#endif
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
||||||
class LstmCPUKernel : public MKLCPUKernel {
|
class LstmCPUKernel : public MKLCPUKernel {
|
|
@ -13,14 +13,14 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/lstm_grad_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/lstm_grad_cpu_kernel.h"
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,12 +13,12 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/matmul_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/matmul_cpu_kernel.h"
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,12 +13,12 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -21,8 +21,8 @@
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include "dnnl.hpp"
|
#include "dnnl.hpp"
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,7 +13,7 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "utils/log_adapter.h"
|
#include "utils/log_adapter.h"
|
||||||
#include "dnnl.hpp"
|
#include "dnnl.hpp"
|
||||||
|
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/mul_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mul_cpu_kernel.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,12 +13,12 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/pooling_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/pooling_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,13 +13,13 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/pooling_grad_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/pooling_grad_cpu_kernel.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -19,7 +19,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/relu_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/relu_cpu_kernel.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/relu_grad_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/relu_grad_cpu_kernel.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/softmax_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/softmax_cpu_kernel.h"
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,12 +13,12 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/softmax_cross_entropy_with_logits_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/softmax_cross_entropy_with_logits_cpu_kernel.h"
|
||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,12 +13,12 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/mkldnn/sparse_softmax_cross_entropy_with_logits_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/sparse_softmax_cross_entropy_with_logits_cpu_kernel.h"
|
||||||
#include <numeric>
|
#include <numeric>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
#include "kernel/cpu/mkldnn/mkl_kernel_engine.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
#include "common/utils.h"
|
#include "common/utils.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
|
@ -18,7 +18,7 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/mkldnn/mkl_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,8 +13,8 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/one_hot_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/one_hot_cpu_kernel.h"
|
||||||
#include "device/cpu/cpu_device_address.h"
|
#include "runtime/device/cpu/cpu_device_address.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -17,8 +17,8 @@
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_ONE_HOT_CPU_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_ONE_HOT_CPU_KERNEL_H_
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel.h"
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,7 +13,7 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/ps/apply_momentum_ps_kernel.h"
|
#include "backend/kernel_compiler/cpu/ps/apply_momentum_ps_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,8 +18,8 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/ps/pserver_kernel.h"
|
#include "backend/kernel_compiler/cpu/ps/pserver_kernel.h"
|
||||||
#include "kernel/cpu/apply_momentum_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/apply_momentum_cpu_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -13,9 +13,9 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
#include "kernel/cpu/ps/embedding_look_up_proxy_kernel.h"
|
#include "backend/kernel_compiler/cpu/ps/embedding_look_up_proxy_kernel.h"
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include "parallel/ps/worker.h"
|
#include "frontend/parallel/ps/worker.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -16,9 +16,9 @@
|
||||||
#ifndef MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_PROXY_KERNEL_H_
|
#ifndef MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_PROXY_KERNEL_H_
|
||||||
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_PROXY_KERNEL_H_
|
#define MINDSPORE_CCSRC_KERNEL_CPU_EMBEDDING_LOOK_UP_PROXY_KERNEL_H_
|
||||||
|
|
||||||
#include "kernel/cpu/embedding_look_up_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/embedding_look_up_cpu_kernel.h"
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include "kernel/cpu/cpu_kernel_factory.h"
|
#include "backend/kernel_compiler/cpu/cpu_kernel_factory.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -14,12 +14,12 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "kernel/cpu/ps/embedding_look_up_ps_kernel.h"
|
#include "backend/kernel_compiler/cpu/ps/embedding_look_up_ps_kernel.h"
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/common_utils.h"
|
#include "backend/kernel_compiler/common_utils.h"
|
||||||
#include "parallel/ps/util.h"
|
#include "frontend/parallel/ps/util.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
|
@ -18,8 +18,8 @@
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "kernel/cpu/embedding_look_up_cpu_kernel.h"
|
#include "backend/kernel_compiler/cpu/embedding_look_up_cpu_kernel.h"
|
||||||
#include "kernel/cpu/ps/pserver_kernel.h"
|
#include "backend/kernel_compiler/cpu/ps/pserver_kernel.h"
|
||||||
|
|
||||||
namespace mindspore {
|
namespace mindspore {
|
||||||
namespace kernel {
|
namespace kernel {
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue