diff options
| author | soumith <soumith@fb.com> | 2016-12-03 03:33:27 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2016-12-03 03:33:27 +0300 |
| commit | 35b9fff3e1fb4d0a379f380319ad5686148d5bcf (patch) | |
| tree | a81f0f4b815679a202b367bcd1b43b914460e507 | |
| parent | b3b9a88aea184c6ae63a03647caaa1c2cc6dc732 (diff) | |
fix half type numerics issue in SpatialFractionalMaxPoolingspatfracmax
| -rw-r--r-- | lib/THCUNN/SpatialFractionalMaxPooling.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/SpatialFractionalMaxPooling.cu b/lib/THCUNN/SpatialFractionalMaxPooling.cu index 22a2878..f3ca162 100644 --- a/lib/THCUNN/SpatialFractionalMaxPooling.cu +++ b/lib/THCUNN/SpatialFractionalMaxPooling.cu @@ -74,7 +74,7 @@ __global__ void SpatialFractionalMaxPooling_updateOutput( } } - assert(maxVal != THCNumerics<Dtype>::min()); + assert(THCNumerics<Dtype>::ne(maxVal, THCNumerics<Dtype>::min())); assert(maxIndex != -1); // +1 for Lua index |