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

!40990 dynamicshape fixbug 

Merge pull request !40990 from TuDouNi/codeclean
This commit is contained in:
i-robot 2022-08-27 05:52:49 +00:00 committed by Gitee
parent 5682cdbec7 a6649edc8b
commit cf616c0f05
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/plugin/device/ascend/kernel/host/dynamic_shape_kernel.cc
View File

@ -23,7 +23,7 @@
namespace mindspore {
namespace kernel {
void TensorShapeKernelMod::Execute() const {
void TensorShapeKernelMod::Execute(void *stream_ptr) const {
MS_LOG(INFO) << "Execute TensorShapeKernel Start";
auto node = anf_node_.lock();
MS_EXCEPTION_IF_NULL(node);
@ -59,10 +59,10 @@ void TensorShapeKernelMod::Execute() const {
}
} else {
// cppcheck-suppress unreadVariable
auto lock = device::KernelRuntime::LockRuntime(stream_);
auto lock = device::KernelRuntime::LockRuntime(stream_ptr);
auto status =
rtMemcpyAsync(const_cast<void *>(output_addr->GetPtr()), output_addr->GetSize(), output_tensor_for_sync->data_c(),
LongToSize(output_tensor_for_sync->data().nbytes()), RT_MEMCPY_HOST_TO_DEVICE_EX, stream_);
LongToSize(output_tensor_for_sync->data().nbytes()), RT_MEMCPY_HOST_TO_DEVICE_EX, stream_ptr);
if (status != RT_ERROR_NONE) {
MS_LOG(EXCEPTION) << "Execute TensorShapeKernel rtMemcpyAsync failed!";
}
@ -78,7 +78,7 @@ bool TensorShapeKernelMod::Launch(const std::vector<AddressPtr> &, const std::ve
auto cnode = node->cast<CNodePtr>();
MS_EXCEPTION_IF_NULL(cnode);
try {
Execute();
Execute(stream_ptr);
} catch (const std::exception &e) {
MS_LOG(ERROR) << "TensorShapeKernelMod Launch failed. node: " << cnode->fullname_with_scope()
<< ", Error message is " << e.what();

2
mindspore/ccsrc/plugin/device/ascend/kernel/host/dynamic_shape_kernel.h
View File

@ -30,7 +30,7 @@ class TensorShapeKernelMod : public HostKernelMod {
const std::vector<AddressPtr> &outputs, void *stream_ptr) override;
private:
void Execute() const;
void Execute(void *stream_ptr) const;
};
MS_HOST_REG_KERNEL(DynamicShape, TensorShapeKernelMod);
MS_HOST_REG_KERNEL(TensorShape, TensorShapeKernelMod);