matrix_relu_derivative.cu File Reference

Implementation of the ReLU derivative function for matrices. More...

#include "matrix.h"
#include <cuda_runtime.h>

Include dependency graph for matrix_relu_derivative.cu:

Go to the source code of this file.

Functions
__global__ void	reluDerivativeKernel (const double *input, double *output, int size)
	CUDA kernel for applying the ReLU derivative function element-wise. More...

Detailed Description

Implementation of the ReLU derivative function for matrices.

Definition in file matrix_relu_derivative.cu.

Function Documentation

reluDerivativeKernel()

__global__ void reluDerivativeKernel	(	const double *	input,
		double *	output,
		int	size
	)

CUDA kernel for applying the ReLU derivative function element-wise.

Parameters

input	Pointer to the input matrix data.
output	Pointer to the output matrix data.
size	Total number of elements in the matrix.

Definition at line 15 of file matrix_relu_derivative.cu.

                                                                                    {
    // Calculate the global thread index
    int idx = blockIdx.x * blockDim.x + threadIdx.x;
    
    // Check if the thread is within the matrix bounds
    if (idx < size) {
        // Apply ReLU derivative: 1 if x > 0, 0 otherwise
        output[idx] = (input[idx] > 0.0) ? 1.0 : 0.0;
    }
}