Skip to content

Commit a348a91

Browse files
committed
Minor fixes
1 parent 9371b23 commit a348a91

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

quaddtype/quaddtype/src/umath.c

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ quad_multiply_strided_loop(PyArrayMethod_Context *context, char *const data[],
2525
char *in1 = data[0], *in2 = data[1];
2626
char *out = data[2];
2727
npy_intp in1_stride = strides[0];
28-
npy_intp in2_stride = strides[0];
28+
npy_intp in2_stride = strides[1];
2929
npy_intp out_stride = strides[2];
3030

3131
while (N--) {
@@ -56,9 +56,18 @@ quad_multiply_resolve_descriptors(PyObject *self, PyArray_DTypeMeta *dtypes[],
5656
// The operand units can be used as-is; no casting required for quad types.
5757
Py_INCREF(given_descrs[0]);
5858
loop_descrs[0] = given_descrs[0];
59+
60+
if (given_descrs[1] == NULL) {
61+
Py_INCREF(given_descrs[0]);
62+
loop_descrs[1] = given_descrs[0];
63+
return NPY_NO_CASTING;
64+
}
65+
Py_INCREF(given_descrs[1]);
66+
loop_descrs[1] = given_descrs[1];
67+
5968
Py_INCREF(given_descrs[1]);
6069
loop_descrs[1] = given_descrs[1];
61-
return NPY_NO_CASTING;
70+
return NPY_SAFE_CASTING;
6271
}
6372

6473
// Function that adds our multiply loop to NumPy's multiply ufunc.

0 commit comments

Comments
 (0)