forked from mindspore-Ecosystem/mindspore
upgrade Ascend software package 17 Sep 21
This commit is contained in:
parent
f4cb1b7586
commit
012305fc43
|
@ -1 +1 @@
|
|||
Subproject commit 329eb8bbed9137a1ed6aca07ec68cdee6ae2975f
|
||||
Subproject commit 15116d1c0cc418fc4dbe3792bfde12811721d58a
|
|
@ -245,10 +245,13 @@ KernelPackPtr TbeUtils::InsertCache(const std::string &kernel_name, const std::s
|
|||
|
||||
int KernelManager::BinaryRegister(const mindspore::kernel::FlexArray &kernel_buffer, void **module, const string &magic,
|
||||
const bool dynamic_flag) {
|
||||
static std::map<string, uint32_t> magic_maps = {{"RT_DEV_BINARY_MAGIC_ELF", RT_DEV_BINARY_MAGIC_ELF},
|
||||
{"RT_DEV_BINARY_MAGIC_PLAIN", RT_DEV_BINARY_MAGIC_PLAIN},
|
||||
static std::map<string, uint32_t> magic_maps = {{"RT_DEV_BINARY_MAGIC_PLAIN", RT_DEV_BINARY_MAGIC_PLAIN},
|
||||
{"RT_DEV_BINARY_MAGIC_PLAIN_AICPU", RT_DEV_BINARY_MAGIC_PLAIN_AICPU},
|
||||
{"RT_DEV_BINARY_MAGIC_ELF_AICPU", RT_DEV_BINARY_MAGIC_ELF_AICPU}};
|
||||
{"RT_DEV_BINARY_MAGIC_PLAIN_AIVEC", RT_DEV_BINARY_MAGIC_PLAIN_AIVEC},
|
||||
{"RT_DEV_BINARY_MAGIC_ELF", RT_DEV_BINARY_MAGIC_ELF},
|
||||
{"RT_DEV_BINARY_MAGIC_ELF_AICPU", RT_DEV_BINARY_MAGIC_ELF_AICPU},
|
||||
{"RT_DEV_BINARY_MAGIC_ELF_AIVEC", RT_DEV_BINARY_MAGIC_ELF_AIVEC},
|
||||
{"RT_DEV_BINARY_MAGIC_ELF_AICUBE", RT_DEV_BINARY_MAGIC_ELF_AICUBE}};
|
||||
// object for device register.
|
||||
rtDevBinary_t dev_bin;
|
||||
dev_bin.data = kernel_buffer.contents;
|
||||
|
|
|
@ -1 +1 @@
|
|||
1.5.0
|
||||
1.5.0rc1
|
Loading…
Reference in New Issue