diff --git a/alias_free_activation/cuda/anti_alias_activation_cuda.cu b/alias_free_activation/cuda/anti_alias_activation_cuda.cu index 8c44233..c22dc21 100644 --- a/alias_free_activation/cuda/anti_alias_activation_cuda.cu +++ b/alias_free_activation/cuda/anti_alias_activation_cuda.cu @@ -57,7 +57,7 @@ namespace input_t down_filter[FILTER_SIZE]; // Load data from global memory including extra indices reserved for replication paddings - input_t elements[2 * FILTER_SIZE + 2 * BUFFER_SIZE + 2 * UPSAMPLE_REPLICATION_PAD] = {0}; + input_t elements[2 * FILTER_SIZE + 2 * BUFFER_SIZE + 2 * UPSAMPLE_REPLICATION_PAD + 1] = {0}; input_t intermediates[2 * FILTER_SIZE + 2 * BUFFER_SIZE + DOWNSAMPLE_REPLICATION_PAD_LEFT + DOWNSAMPLE_REPLICATION_PAD_RIGHT] = {0}; // Output stores downsampled output before writing to dst