audio_conv1d_checked_internal.h File Reference

#include <stddef.h>

Go to the source code of this file.

Data Structures
struct	ck_audio_conv1d_job

Typedefs
typedef void(*	ck_audio_conv1d_compute_fn) (const ck_audio_conv1d_job *, void *)

Functions
void	audio_conv1d_compute_avx2 (const ck_audio_conv1d_job *job, void *unused)
int	ck_audio_conv1d_checked_execute (const float *input, size_t input_elements, size_t input_stride, const float *weight, size_t weight_elements, const float *bias, size_t bias_elements, float *output, size_t output_elements, size_t output_stride, float *scratch, size_t scratch_elements, size_t input_channels, size_t output_channels, size_t input_frames, size_t kernel_size, size_t stride, size_t padding, size_t dilation, size_t output_frames, ck_audio_conv1d_compute_fn compute, void *compute_context)

Typedef Documentation

ck_audio_conv1d_compute_fn

typedef void(* ck_audio_conv1d_compute_fn) (const ck_audio_conv1d_job *, void *)

Definition at line 24 of file audio_conv1d_checked_internal.h.

Function Documentation

audio_conv1d_compute_avx2()

void audio_conv1d_compute_avx2	(	const ck_audio_conv1d_job *	job,
		void *	unused
	)

Referenced by audio_conv1d_checked_channel_major_f32_avx2_fma(), and audio_conv1d_dilated_checked_channel_major_f32_avx2_fma().

ck_audio_conv1d_checked_execute()

int ck_audio_conv1d_checked_execute	(	const float *	input,
		size_t	input_elements,
		size_t	input_stride,
		const float *	weight,
		size_t	weight_elements,
		const float *	bias,
		size_t	bias_elements,
		float *	output,
		size_t	output_elements,
		size_t	output_stride,
		float *	scratch,
		size_t	scratch_elements,
		size_t	input_channels,
		size_t	output_channels,
		size_t	input_frames,
		size_t	kernel_size,
		size_t	stride,
		size_t	padding,
		size_t	dilation,
		size_t	output_frames,
		ck_audio_conv1d_compute_fn	compute,
		void *	compute_context
	)

Definition at line 142 of file audio_conv1d_checked.c.

                                                               {
    size_t input_span, output_span, weight_count, scratch_need;
    if (!input_channels || !output_channels || !input_frames ||
        !kernel_size || !stride || !dilation || !output_frames ||
        input_channels > INT_MAX || output_channels > INT_MAX ||
        input_frames > INT_MAX || kernel_size > INT_MAX ||
        stride > INT_MAX || padding > INT_MAX || dilation > INT_MAX ||
        output_frames > INT_MAX)
        return CK_AUDIO_CONV_CHECKED_INVALID;
    const int64_t effective_kernel = 1 +
        ((int64_t)kernel_size - 1) * (int64_t)dilation;
    const int64_t padded = (int64_t)input_frames +
        2 * (int64_t)padding - effective_kernel;
    if (padded < 0 || padded / stride + 1 != output_frames)
        return CK_AUDIO_CONV_CHECKED_INVALID;
    if (!ck_checked_span((size_t)input_channels, (size_t)input_frames,
                         input_stride, &input_span) ||
        !ck_checked_span((size_t)output_channels, (size_t)output_frames,
                         output_stride, &output_span) ||
        audio_conv1d_checked_workspace((size_t)input_channels,
            (size_t)output_channels, (size_t)input_frames,
            (size_t)output_frames, &scratch_need))
        return CK_AUDIO_CONV_CHECKED_OVERFLOW;
    if ((size_t)output_channels > SIZE_MAX / (size_t)input_channels ||
        (size_t)output_channels * (size_t)input_channels > SIZE_MAX / (size_t)kernel_size)
        return CK_AUDIO_CONV_CHECKED_OVERFLOW;
    weight_count = (size_t)output_channels * (size_t)input_channels * (size_t)kernel_size;
    if (weight_count > SIZE_MAX / sizeof(float)) return CK_AUDIO_CONV_CHECKED_OVERFLOW;
    if (input_elements < input_span || weight_elements < weight_count ||
        bias_elements < (size_t)output_channels || output_elements < output_span ||
        scratch_elements < scratch_need)
        return CK_AUDIO_CONV_CHECKED_CAPACITY;
    if (!ck_checked_region(input, input_span) ||
        !ck_checked_region(weight, weight_count) ||
        !ck_checked_region(bias, (size_t)output_channels) ||
        !ck_checked_region(output, output_span) ||
        !ck_checked_region(scratch, scratch_need))
        return CK_AUDIO_CONV_CHECKED_INVALID;
    if (ck_checked_overlap(output, output_span, input, input_span) ||
        ck_checked_overlap(output, output_span, weight, weight_count) ||
        ck_checked_overlap(output, output_span, bias, (size_t)output_channels) ||
        ck_checked_overlap(output, output_span, scratch, scratch_need) ||
        ck_checked_overlap(scratch, scratch_need, input, input_span) ||
        ck_checked_overlap(scratch, scratch_need, weight, weight_count) ||
        ck_checked_overlap(scratch, scratch_need, bias, (size_t)output_channels))
        return CK_AUDIO_CONV_CHECKED_INVALID;
    if (!ck_checked_finite(weight, weight_count) ||
        !ck_checked_finite(bias, (size_t)output_channels))
        return CK_AUDIO_CONV_CHECKED_NONFINITE;
    for (size_t channel = 0; channel < input_channels; ++channel)
        if (!ck_checked_finite(input + (size_t)channel * input_stride,
                               (size_t)input_frames))
            return CK_AUDIO_CONV_CHECKED_NONFINITE;
    /* Validation and staged commit remain shared by both providers. */
    const ck_audio_conv1d_job job = {input, weight, bias, scratch,
        input_stride, input_channels, output_channels, input_frames,
        kernel_size, stride, padding, dilation, output_frames, 0,
        output_channels};
    compute(&job, compute_context);
    if (!ck_checked_finite(scratch, (size_t)output_channels * (size_t)output_frames))
        return CK_AUDIO_CONV_CHECKED_NONFINITE;
    for (size_t channel = 0; channel < output_channels; ++channel)
        memcpy(output + (size_t)channel * output_stride,
               scratch + (size_t)channel * (size_t)output_frames,
               (size_t)output_frames * sizeof(float));
    return CK_AUDIO_CONV_CHECKED_OK;
}

References audio_conv1d_checked_workspace(), CK_AUDIO_CONV_CHECKED_CAPACITY, CK_AUDIO_CONV_CHECKED_INVALID, CK_AUDIO_CONV_CHECKED_NONFINITE, CK_AUDIO_CONV_CHECKED_OK, CK_AUDIO_CONV_CHECKED_OVERFLOW, ck_checked_finite(), ck_checked_overlap(), ck_checked_region(), and ck_checked_span().

Referenced by audio_conv1d_checked_channel_major_f32(), audio_conv1d_checked_channel_major_f32_avx2_fma(), audio_conv1d_checked_channel_major_f32_hwfma(), audio_conv1d_dilated_checked_channel_major_f32(), audio_conv1d_dilated_checked_channel_major_f32_avx2_fma(), audio_conv1d_dilated_checked_channel_major_f32_hwfma(), and audio_conv1d_parallel_checked_impl().