Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!49750 Fix AkgAscendKernelMod Bug 

Merge pull request !49750 from jiaoy1224/fixbug
This commit is contained in:
i-robot 2023-03-06 09:32:09 +00:00 committed by Gitee
parent ecada229a8 4c2ba4a59f
commit a06ed62f95
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/ascend/kernel/akg/akg_ascend_kernel_build.cc
View File

@ -35,7 +35,7 @@ KernelPackPtr AkgAscendKernelBuilder::AkgInsertCache(const std::string &kernel_n
void AkgAscendKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack,
const AkgKernelJsonGenerator &json_generator, const AnfNodePtr &anf_node) {
auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack);
auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack, anf_node);
auto kernel_json_info = kernel_pack->kernel_json_info();
kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list());
kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list());

3
mindspore/ccsrc/plugin/device/ascend/kernel/akg/akg_ascend_kernel_mod.cc
View File

@ -37,7 +37,8 @@ constexpr size_t ARGS_REMAP_LEN = 2;
/**
* @brief infotable contain func_stub\blockdim\kernel file buffer
*/
AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {
AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack, const AnfNodePtr &anf_node_ptr)
: AscendKernelMod(anf_node_ptr), kernel_pack_(kernel_pack) {
if (kernel_pack != nullptr) {
auto kernel_json_info = kernel_pack->kernel_json_info();
kernel_name_ = kernel_json_info.kernel_name;

2
mindspore/ccsrc/plugin/device/ascend/kernel/akg/akg_ascend_kernel_mod.h
View File

@ -26,7 +26,7 @@ namespace mindspore {
namespace kernel {
class AkgKernelMod : public AscendKernelMod {
public:
explicit AkgKernelMod(const KernelPackPtr &kernel_pack);
explicit AkgKernelMod(const KernelPackPtr &kernel_pack, const AnfNodePtr &anf_node_ptr);
~AkgKernelMod() final {}
bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
const std::vector<AddressPtr> &outputs, void *stream_ptr) override;