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

apply power sign clean code

This commit is contained in:
chujinjin 2022-11-01 11:28:53 +08:00
parent 482ec11586
commit 55003fe07e
3 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/plugin/device/cpu/kernel/apply_power_sign_cpu_kernel.cc
View File

@ -83,6 +83,7 @@ void ApplyPowerSignCpuKernelMod::LaunchPowerSign(const std::vector<kernel::Addre
bool ApplyPowerSignCpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
const std::vector<KernelTensorPtr> &outputs) {
MS_EXCEPTION_IF_NULL(base_operator);
kernel_name_ = base_operator->name();
dtype_ = inputs[0]->GetDtype();
batch_rank_ = base_operator->get_batch_rank();

3
mindspore/ccsrc/plugin/device/cpu/kernel/scale_grad_cpu_kernel.cc
View File

@ -54,6 +54,7 @@ void ScaleGradCpuKernelMod::LaunchScaleGradPerGrad(const std::vector<AddressPtr>
if (scale_addr_half != nullptr) {
x1 = static_cast<T>(*scale_addr_half);
} else {
MS_EXCEPTION_IF_NULL(scale_addr_float);
x1 = static_cast<T>(*scale_addr_float);
}
@ -101,7 +102,9 @@ std::vector<KernelAttr> ScaleGradCpuKernelMod::GetOpSupport() {
bool ScaleGradCpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
const std::vector<KernelTensorPtr> &outputs) {
MS_EXCEPTION_IF_NULL(base_operator);
auto kernel_ptr = std::dynamic_pointer_cast<ops::ScaleGrad>(base_operator);
MS_EXCEPTION_IF_NULL(kernel_ptr);
kernel_name_ = kernel_ptr->name();
auto input_size = inputs.size();
for (size_t index = 0; index < input_size; index++) {

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

@ -50,6 +50,7 @@ void ScaleGradGpuKernelMod::LaunchScaleGradPerGrad(const std::vector<AddressPtr>
ScaleGradKernel(outputs[index]->size / sizeof(T), input_addr, *scale_addr_half, output_addr,
reinterpret_cast<cudaStream_t>(stream_ptr));
} else {
MS_EXCEPTION_IF_NULL(scale_addr_float);
ScaleGradKernel(outputs[index]->size / sizeof(T), input_addr, *scale_addr_float, output_addr,
reinterpret_cast<cudaStream_t>(stream_ptr));
}
@ -84,7 +85,9 @@ bool ScaleGradGpuKernelMod::Launch(const std::vector<AddressPtr> &inputs, const
bool ScaleGradGpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
const std::vector<KernelTensorPtr> &outputs) {
MS_EXCEPTION_IF_NULL(base_operator);
auto kernel_ptr = std::dynamic_pointer_cast<ops::ScaleGrad>(base_operator);
MS_EXCEPTION_IF_NULL(kernel_ptr);
kernel_name_ = kernel_ptr->name();
auto input_size = inputs.size();
for (size_t index = 0; index < input_size; index++) {