ndarray: protect from diving zero for non-inplace op (#378)

dmlc · Dec 15, 2017 · 1e20f50 · 1e20f50
1 parent 8135a63
commit 1e20f50
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/ndarray.jl b/src/ndarray.jl
@@ -717,6 +717,11 @@ broadcast_(::typeof(/), x::NDArray, y::NDArray) = _div(x, y)
 broadcast_(::typeof(/), x::NDArray, y::Real)    = _div_scalar(x, scalar = y)
 broadcast_(::typeof(/), y::Real, x::NDArray)    = _rdiv_scalar(x, scalar = y)
 
+function broadcast_(::typeof(/), x::NDArray{T}, y::Real) where {T<:Integer}
+  @assert(round(T, y) != zero(T), "Integer divided by zero")
+  _div_scalar(x, scalar = y)
+end
+
 import Base: %
 
 """