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

codex

This commit is contained in:
gongdaguo1 2022-09-26 10:35:26 +08:00
parent 779278dbbf
commit b08aca37a7
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/plugin/device/gpu/kernel/arrays/masked_select_grad_gpu_kernel.cc
View File

@ -89,7 +89,6 @@ void MaskedSelectGradGpuKernelMod::ResetResource() noexcept {
input_size_ = 0;
mask_size_ = 0;
broadcast_size_ = 0;
// real_output_size_ = 0;
input_broadcast_ = false;
mask_broadcast_ = false;
for (size_t i = 0; i < MAX_DIMS; i++) {
@ -163,7 +162,6 @@ int MaskedSelectGradGpuKernelMod::Resize(const BaseOperatorPtr &base_operator,
if (mask_broadcast_) {
workspace_size_list_.push_back(broadcast_size_ * mask_type_size_); // save broadcast result of mask
}
// output_size_list_.push_back(broadcast_size_ * input_type_size_);
input_size_list_.push_back(broadcast_size_ * input_type_size_);
output_size_list_.push_back(input_size_ * input_type_size_);
return KRET_OK;
@ -198,7 +196,6 @@ bool MaskedSelectGradGpuKernelMod::LaunchKernel(const std::vector<AddressPtr> &i
mask_broadcast_ptr = GetDeviceAddress<bool>(workspace, kIndex1);
MS_EXCEPTION_IF_NULL(mask_broadcast_ptr);
}
// auto output_ptr = GetDeviceAddress<T>(outputs, kIndex0);
auto output_grad_ptr = GetDeviceAddress<T>(inputs, kIndexOutputGrad);
if (output_grad_ptr == nullptr) {
auto input_grad_ptr = GetDeviceAddress<T>(outputs, kIndex0);

3
mindspore/core/ops/grad/masked_select_grad.h
View File

@ -16,7 +16,6 @@
#ifndef MINDSPORE_CORE_OPS_MASKEDGRAD_H_
#define MINDSPORE_CORE_OPS_MASKEDGRAD_H_
#include <map>
#include <vector>
#include <string>
#include <memory>
#include "ops/base_operator.h"
@ -29,7 +28,7 @@ class MIND_API MaskedSelectGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(MaskedSelectGrad);
MaskedSelectGrad() : BaseOperator(kNameMaskedSelectGrad) {}
void Init() {}
void Init() const {}
};
} // namespace ops
} // namespace mindspore