Rollback of #231

PiperOrigin-RevId: 694235313
jax-ml · Nov 7, 2024 · 46430e6 · 46430e6
1 parent e89cc9d
commit 46430e6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ml_dtypes/_src/custom_float.h b/ml_dtypes/_src/custom_float.h
@@ -401,7 +401,7 @@ PyArray_DescrProto GetCustomFloatDescrProto() {
       /*kind=*/TypeDescriptor<T>::kNpyDescrKind,
       /*type=*/TypeDescriptor<T>::kNpyDescrType,
       /*byteorder=*/TypeDescriptor<T>::kNpyDescrByteorder,
-      /*flags=*/NPY_USE_SETITEM,
+      /*flags=*/NPY_NEEDS_PYAPI | NPY_USE_SETITEM,
       /*type_num=*/0,
       /*elsize=*/sizeof(T),
       /*alignment=*/alignof(T),

diff --git a/ml_dtypes/_src/intn_numpy.h b/ml_dtypes/_src/intn_numpy.h
@@ -413,7 +413,7 @@ PyArray_DescrProto GetIntNDescrProto() {
       /*kind=*/TypeDescriptor<T>::kNpyDescrKind,
       /*type=*/TypeDescriptor<T>::kNpyDescrType,
       /*byteorder=*/TypeDescriptor<T>::kNpyDescrByteorder,
-      /*flags=*/NPY_USE_SETITEM,
+      /*flags=*/NPY_NEEDS_PYAPI | NPY_USE_SETITEM,
       /*type_num=*/0,
       /*elsize=*/sizeof(T),
       /*alignment=*/alignof(T),