Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/cunn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGregory Chanan <gchanan@fb.com>2016-11-03 19:22:27 +0300
committerGregory Chanan <gchanan@fb.com>2016-11-09 00:46:52 +0300
commitb0444c9da3367e44a0de17d8250742906b47ce46 (patch)
treeef9380d5e5ad0af47493ae2be1e3e6d99a7e4b9b /lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu
parent9fd7a885be5db599adcb868494632a3fe0de0205 (diff)
ArgCheck that dilation parameters are > 0 and ensure tests
pick dilation parameters > 0.
Diffstat (limited to 'lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu')
-rw-r--r--lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu3
1 files changed, 3 insertions, 0 deletions
diff --git a/lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu b/lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu
index bd43bc0..0d6d084 100644
--- a/lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu
+++ b/lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu
@@ -71,6 +71,9 @@ void THNN_(VolumetricDilatedMaxPooling_updateOutput)(
THArgCheck(kT/2 >= padT && kW/2 >= padW && kH/2 >= padH, 2,
"pad should be smaller than half of kernel size"
);
+ THArgCheck(dilationT > 0 && dilationW > 0 && dilationH > 0, 14,
+ "dilation should be greater than 0"
+ );
if (ceilMode)
{