drop duplicate error handling

Author: casperdcl

numcu/src/numcu.cu

@@ -12,52 +12,20 @@ namespace py = pybind11;
 template <typename T>
 void elem_div(py::buffer num, py::buffer den, py::buffer dst, T zeroDivDefault) {
   py::buffer_info src_num = num.request(), src_den = den.request(), dst_out = dst.request(true);
-  const std::vector<std::string> types = {src_num.format, src_den.format, dst_out.format};
-  for (auto &type : types) {
-    if (type != py::format_descriptor<T>::format()) throw py::type_error("unexpected type");
-  }
-  if (src_num.ndim != src_den.ndim) throw py::index_error("inputs must have same ndim");
-  if (src_num.ndim != dst_out.ndim) throw py::index_error("output must have same ndim");
-  for (size_t i = 0; i < src_num.ndim; i++) {
-    if (src_num.shape[i] != src_den.shape[i]) throw py::index_error("inputs must have same shape");
-    if (src_num.shape[i] != dst_out.shape[i]) throw py::index_error("output must have same shape");
-  }
-
   div(static_cast<T *>(dst_out.ptr), static_cast<T *>(src_num.ptr), static_cast<T *>(src_den.ptr),
       dst_out.size, zeroDivDefault);
   CUDA_PyErr();
 }
 
 template <typename T> void elem_mul(py::buffer a, py::buffer b, py::buffer dst) {
   py::buffer_info src_a = a.request(), src_b = b.request(), dst_out = dst.request(true);
-  const std::vector<std::string> types = {src_a.format, src_b.format, dst_out.format};
-  for (auto &type : types) {
-    if (type != py::format_descriptor<T>::format()) throw py::type_error("unexpected type");
-  }
-  if (src_a.ndim != src_b.ndim) throw py::index_error("inputs must have same ndim");
-  if (src_a.ndim != dst_out.ndim) throw py::index_error("output must have same ndim");
-  for (size_t i = 0; i < src_a.ndim; i++) {
-    if (src_a.shape[i] != src_b.shape[i]) throw py::index_error("inputs must have same shape");
-    if (src_a.shape[i] != dst_out.shape[i]) throw py::index_error("output must have same shape");
-  }
-
   mul(static_cast<T *>(dst_out.ptr), static_cast<T *>(src_a.ptr), static_cast<T *>(src_b.ptr),
       dst_out.size);
   CUDA_PyErr();
 }
 
 template <typename T> void elem_add(py::buffer a, py::buffer b, py::buffer dst) {
   py::buffer_info src_a = a.request(), src_b = b.request(), dst_out = dst.request(true);
-  const std::vector<std::string> types = {src_a.format, src_b.format, dst_out.format};
-  for (auto &type : types) {
-    if (type != py::format_descriptor<T>::format()) throw py::type_error("unexpected type");
-  }
-  if (src_a.ndim != src_b.ndim) throw py::index_error("inputs must have same ndim");
-  if (src_a.ndim != dst_out.ndim) throw py::index_error("output must have same ndim");
-  for (size_t i = 0; i < src_a.ndim; i++) {
-    if (src_a.shape[i] != src_b.shape[i]) throw py::index_error("inputs must have same shape");
-    if (src_a.shape[i] != dst_out.shape[i]) throw py::index_error("output must have same shape");
-  }
   add(static_cast<T *>(dst_out.ptr), static_cast<T *>(src_a.ptr), static_cast<T *>(src_b.ptr),
       dst_out.size);
   CUDA_PyErr();