diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-11-03 19:22:27 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-11-09 00:46:52 +0300 |
commit | b0444c9da3367e44a0de17d8250742906b47ce46 (patch) | |
tree | ef9380d5e5ad0af47493ae2be1e3e6d99a7e4b9b /lib/THCUNN/generic/VolumetricDilatedMaxPooling.cu | |
parent | 9fd7a885be5db599adcb868494632a3fe0de0205 (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.cu | 3 |
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) { |