forked from OSSInnovation/mindspore
add loss cuda
This commit is contained in:
parent
2eefb94fee
commit
2e8f37cc5d
|
|
@ -0,0 +1,198 @@
|
|||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include <algorithm>
|
||||
#include "loss_with_reduction_impl.cuh"
|
||||
#include "runtime/device/gpu/cuda_common.h"
|
||||
|
||||
template <typename T>
|
||||
__global__ void Copy(T *loss, T *tmp_loss, int reduction, int input_size) {
|
||||
loss[0] += tmp_loss[0];
|
||||
if (reduction == 1) {
|
||||
loss[0] /= input_size;
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void AddTile(T *tmp_loss, int index) {
|
||||
tmp_loss[0] += tmp_loss[index];
|
||||
}
|
||||
template <typename T>
|
||||
__global__ void PartialSum(T *tmp_loss, int stride) {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < stride; i += blockDim.x * gridDim.x) {
|
||||
tmp_loss[i] += tmp_loss[i + stride];
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void LossInitKernel(T *loss) {
|
||||
loss[0] = static_cast<T>(0.);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void KLDivLossKernel(const int input_size, const int reduction, const T *input_x, const T *input_y, T *loss,
|
||||
T *tmp_loss) {
|
||||
T epsilon = 1e-6;
|
||||
if (reduction == 0) {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_y[i], epsilon);
|
||||
T value = input_y[i] * (logf(denominator) - input_x[i]);
|
||||
loss[i] = value;
|
||||
}
|
||||
} else {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_y[i], epsilon);
|
||||
T value = input_y[i] * (logf(denominator) - input_x[i]);
|
||||
tmp_loss[i] = value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *loss,
|
||||
cudaStream_t stream) {
|
||||
LossInitKernel<<<1, 1, 0, stream>>>(loss);
|
||||
T *tmp_loss;
|
||||
if (reduction != 0) {
|
||||
cudaMalloc(reinterpret_cast<void **>(&tmp_loss), input_size * sizeof(T));
|
||||
}
|
||||
KLDivLossKernel<<<GET_BLOCKS(input_size), GET_THREADS, 0, stream>>>(input_size, reduction, input_x, input_y, loss,
|
||||
tmp_loss);
|
||||
if (reduction != 0) {
|
||||
if (input_size % 2 == 1) {
|
||||
AddTile<<<1, 1, 0, stream>>>(tmp_loss, input_size - 1);
|
||||
}
|
||||
for (int stride = input_size / 2; stride > 0; stride >>= 1) {
|
||||
PartialSum<<<GET_BLOCKS(stride), GET_THREADS, 0, stream>>>(tmp_loss, stride);
|
||||
if (stride > 2 && stride % 2 == 1) {
|
||||
AddTile<<<1, 1, 0, stream>>>(tmp_loss, stride - 1);
|
||||
}
|
||||
}
|
||||
Copy<<<1, 1, 0, stream>>>(loss, tmp_loss, reduction, input_size);
|
||||
}
|
||||
cudaFree(tmp_loss);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void KLDivLossGradKernel(const int input_size, const int reduction, const T *input_x, const T *input_y,
|
||||
const T *dloss, T *dx, T *dy) {
|
||||
T epsilon = 1e-6;
|
||||
if (reduction == 0) {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_y[i], epsilon);
|
||||
dx[i] = -input_y[i] * dloss[i];
|
||||
dy[i] = (logf(denominator) + 1 - input_x[i]) * dloss[i];
|
||||
}
|
||||
} else {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_y[i], epsilon);
|
||||
dx[i] = -input_y[i] * dloss[0];
|
||||
dy[i] = (logf(denominator) + 1 - input_x[i]) * dloss[0];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void KLDivLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y, const T *dloss,
|
||||
T *dx, T *dy, cudaStream_t stream) {
|
||||
KLDivLossGradKernel<<<GET_BLOCKS(input_size), GET_THREADS, 0, stream>>>(input_size, reduction, input_x, input_y,
|
||||
dloss, dx, dy);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void BinaryCrossEntropyLossKernel(const int input_size, const int reduction, const T *input_x,
|
||||
const T *input_y, const T *weight, T *loss, T *tmp_loss) {
|
||||
T epsilon = 1e-6;
|
||||
if (reduction == 0) {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T antilogarithm = max(input_x[i], epsilon);
|
||||
T antilogarithm2 = min(1 - input_x[i], 1 - epsilon);
|
||||
T value = -weight[i] * (input_y[i] * logf(antilogarithm) + (1 - input_y[i]) * logf(antilogarithm2));
|
||||
loss[i] = value;
|
||||
}
|
||||
} else {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T antilogarithm = max(input_x[i], epsilon);
|
||||
T antilogarithm2 = min(1 - input_x[i], 1 - epsilon);
|
||||
T value = -weight[i] * (input_y[i] * logf(antilogarithm) + (1 - input_y[i]) * logf(antilogarithm2));
|
||||
tmp_loss[i] = value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void BinaryCrossEntropyLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y,
|
||||
const T *weight, T *loss, cudaStream_t stream) {
|
||||
LossInitKernel<<<1, 1, 0, stream>>>(loss);
|
||||
T *tmp_loss;
|
||||
if (reduction != 0) {
|
||||
cudaMalloc(reinterpret_cast<void **>(&tmp_loss), input_size * sizeof(T));
|
||||
}
|
||||
BinaryCrossEntropyLossKernel<<<GET_BLOCKS(input_size), GET_THREADS, 0, stream>>>(input_size, reduction, input_x,
|
||||
input_y, weight, loss, tmp_loss);
|
||||
if (reduction != 0) {
|
||||
if (input_size % 2 == 1) {
|
||||
AddTile<<<1, 1, 0, stream>>>(tmp_loss, input_size - 1);
|
||||
}
|
||||
for (int stride = input_size / 2; stride > 0; stride >>= 1) {
|
||||
PartialSum<<<GET_BLOCKS(stride), GET_THREADS, 0, stream>>>(tmp_loss, stride);
|
||||
if (stride > 2 && stride % 2 == 1) {
|
||||
AddTile<<<1, 1, 0, stream>>>(tmp_loss, stride - 1);
|
||||
}
|
||||
}
|
||||
Copy<<<1, 1, 0, stream>>>(loss, tmp_loss, reduction, input_size);
|
||||
}
|
||||
cudaFree(tmp_loss);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
__global__ void BinaryCrossEntropyLossGradKernel(const int input_size, const int reduction, const T *input_x,
|
||||
const T *input_y, const T *weight, const T *dloss, T *dx) {
|
||||
T epsilon = 1e-6;
|
||||
if (reduction == 0) {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_x[i] * (1 - input_x[i]), epsilon);
|
||||
T value = weight[i] * (input_x[i] - input_y[i]) / denominator;
|
||||
dx[i] = value * dloss[i];
|
||||
}
|
||||
} else {
|
||||
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < input_size; i += blockDim.x * gridDim.x) {
|
||||
T denominator = max(input_x[i] * (1 - input_x[i]), epsilon);
|
||||
T value = weight[i] * (input_x[i] - input_y[i]) / denominator;
|
||||
dx[i] = value * dloss[0];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void BinaryCrossEntropyLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y,
|
||||
const T *weight, const T *dloss, T *dx, cudaStream_t stream) {
|
||||
BinaryCrossEntropyLossGradKernel<<<GET_BLOCKS(input_size), GET_THREADS, 0, stream>>>(input_size, reduction, input_x,
|
||||
input_y, weight, dloss, dx);
|
||||
}
|
||||
|
||||
template void KLDivLoss(const int &input_size, const int &reduction, const float *input_x, const float *input_y,
|
||||
float *loss, cudaStream_t stream);
|
||||
|
||||
template void KLDivLossGrad(const int &input_size, const int &reduction, const float *input_x, const float *input_y,
|
||||
const float *dloss, float *dx, float *dy, cudaStream_t stream);
|
||||
|
||||
template void BinaryCrossEntropyLoss(const int &input_size, const int &reduction, const float *input_x,
|
||||
const float *input_y, const float *weight, float *loss, cudaStream_t stream);
|
||||
|
||||
template void BinaryCrossEntropyLossGrad(const int &input_size, const int &reduction, const float *input_x,
|
||||
const float *input_y, const float *weight, const float *dloss, float *dx,
|
||||
cudaStream_t stream);
|
||||
|
|
@ -0,0 +1,31 @@
|
|||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_GPU_LOSS_WITH_REDUCTION_IMPL_CUH
|
||||
#define MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_GPU_LOSS_WITH_REDUCTION_IMPL_CUH
|
||||
template <typename T>
|
||||
void BinaryCrossEntropyLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y,
|
||||
const T *weight, T *loss, cudaStream_t stream);
|
||||
template <typename T>
|
||||
void BinaryCrossEntropyLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y,
|
||||
const T *weight, const T *dloss, T *dx, cudaStream_t stream);
|
||||
template <typename T>
|
||||
void KLDivLoss(const int &input_size, const int &reduction, const T *input_x, const T *input_y, T *loss,
|
||||
cudaStream_t stream);
|
||||
template <typename T>
|
||||
void KLDivLossGrad(const int &input_size, const int &reduction, const T *input_x, const T *input_y, const T *dloss,
|
||||
T *dx, T *dy, cudaStream_t stream);
|
||||
#endif // MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_GPU_LOSS_WITH_REDUCTION_IMPL_CUH
|
||||
Loading…
Reference in New Issue