Numpy: 배열과 같은 객체에 대해`np.vdot`에서 충돌이 발생합니다.

에 만든 2019년 08월 10일 · 6코멘트 · 출처: numpy/numpy

numpy.vdot 는 비표준 방식으로 __array__ 를 구현하는 객체를 전달하면 segfault가 발생할 수 있습니다.

segfault는 다음 코드 줄에서 발생합니다.
https://github.com/numpy/numpy/blob/master/numpy/core/src/multiarray/multiarraymodule.c#L2245
PyArray_DescrFromType 가 실패하면 type 가 NULL type 가 될 수 있기 때문입니다.

재현 코드 예 :

import numpy as np

class Foo(object):
     def __array__(self, a):
         return self

np.vdot(Foo(), Foo())

에러 메시지:

역 추적 :

(lldb) bt
* thread #1, queue = 'com.apple.main-thread', stop reason = EXC_BAD_ACCESS (code=1, address=0x0)
  * frame #0: 0x0000000100cef88e _multiarray_umath.cpython-37m-darwin.so`array_vdot + 126
    frame #1: 0x00000001005e0994 python`PyCFunction_Call + 148
    frame #2: 0x0000000100c48dc1 _multiarray_umath.cpython-37m-darwin.so`array_implement_array_function + 305
    frame #3: 0x00000001005e11b3 python`_PyMethodDef_RawFastCallKeywords + 227
    frame #4: 0x00000001005e071c python`_PyCFunction_FastCallKeywords + 44
    frame #5: 0x00000001006b6048 python`call_function + 664
    frame #6: 0x00000001006b2a18 python`_PyEval_EvalFrameDefault + 27080
    frame #7: 0x00000001006b6e45 python`_PyEval_EvalCodeWithName + 2997
    frame #8: 0x00000001005e06d6 python`_PyFunction_FastCallKeywords + 230
    frame #9: 0x00000001006b60d9 python`call_function + 809
    frame #10: 0x00000001006b2987 python`_PyEval_EvalFrameDefault + 26935
    frame #11: 0x00000001006b6e45 python`_PyEval_EvalCodeWithName + 2997
    frame #12: 0x00000001006abfb0 python`PyEval_EvalCode + 48
    frame #13: 0x00000001006ee677 python`PyRun_InteractiveOneObjectEx + 615
    frame #14: 0x00000001006ede4e python`PyRun_InteractiveLoopFlags + 190
    frame #15: 0x00000001006edd5c python`PyRun_AnyFileExFlags + 60
    frame #16: 0x0000000100710c51 python`pymain_main + 7873
    frame #17: 0x000000010071127f python`_Py_UnixMain + 111
    frame #18: 0x00007fff5f6e63d5 libdyld.dylib`start + 1

Numpy / Python 버전 정보 :

나는 이것을 Mac OS X에서 다음과 같이 재현했습니다.

1.17.0 3.7.2 (default, Jan 16 2019, 11:36:28)
[Clang 10.0.0 (clang-1000.11.45.2)]

내부 Linux 빌드에서.

이 문제는 원래 https://github.com/google/jax/issues/1162 로보고되었습니다.

00 - Bug

출처

hawkinsp

모든 6 댓글

흠, vdot 는 출력 dtype이 미리 무엇인지 확인합니다 (내가 생각하기에 괜찮습니다). 그러나이 함수는 실패하고 NPY_NOTYPE 반환 할 수 있으므로 NPY_NOTYPE 이 반환 될 때 오류를 반환해야합니다. 이것은 https://github.com/numpy/numpy/blob/master/numpy/core/src/multiarray/multiarraymodule.c#L2241 (및 다음 줄)에 있으며, 표시되는 오류는 나중에 오류 반환입니다. 그러나 나는 그 함수가 실제로 오류를 반환해서는 안된다고 생각합니다 (오류 검사도 추가해도 괜찮습니다).

홍보하고 싶습니까?

(오류 검사가 누락 된 유사한 용도가 더 있는지 확인하는 것이 좋습니다.)

seberg 에 2019년 08월 10일

안녕하세요,
저는 신규 이민자이며이 문제를 해결하고 싶습니다. 어디서부터 시작해야하나요?
미리 감사드립니다.

Soniyanayak51 에 2019년 08월 20일

개발 워크 플로우 에 대해 읽어보십시오. 그런 다음 master 지점 ( git checkout master; git clean -xfd; python runtests.py )에서 python runtests.py 실행할 수있는 지점에 도달해야합니다.

설정이 완료되면 해킹을 시작할 수 있습니다. 브랜치 생성, 실패한 테스트 추가 ( numpy/core/tests/teste_multiarray.py 의 TestVDot 가 좋은 곳), 위의 힌트에서 C 코드를 수정 해보세요. . 문제가 발생하면 언제든지 도움을 요청하면 문서를 따르고 힌트를 따르기 위해 진정으로 노력했음을 더 많이 보여줄 수 있습니다.

mattip 에 2019년 08월 20일

👍1

@ Soniyanayak51 환영합니다!

@mattip의 말에

rgommers 에 2019년 08월 20일

👍1

@rgommers @mattip PyArray_DescrFromType ()에서 얻은 유형에 대한 검사를 추가했습니다. PyArray_DESCR ()에 대해서도 추가해야하는지 잘 모르겠습니다.
또한 test_multiarray.py에 테스트를 추가하는 경우 test_PyArrayType을 추가하고 유형이 NULL 인 테스트를 추가해야합니까? 테스트를 추가하는 방법을 잘 모르겠습니다.

Soniyanayak51 에 2019년 08월 21일

@rgommers @mattip PyArray_DescrFromType ()에서 얻은 유형에 대한 검사를 추가했습니다. PyArray_DESCR ()에 대해서도 추가해야하는지 잘 모르겠습니다.
또한 test_multiarray.py에 테스트를 추가하는 경우 test_PyArrayType을 추가하고 유형이 NULL 인 테스트를 추가해야합니까? 테스트를 추가하는 방법을 잘 모르겠습니다.

@mattip 이 PR에서 부분적으로 대답 한 것 같습니다. 대화를 계속 유지하는 것이 좋습니다.

test_PyArrayType을 추가하고 유형이 NULL 인 테스트를 추가해야합니까?

약간 추가하려면 일반적으로 테스트를 추가하는 원칙은 공용 API 만 테스트하는 것입니다. 그래서 당신은

...
np.vdot(Foo(), Foo())

위의 문제 설명에서 테스트로 추가하십시오. C API를 별도로 테스트 할 필요가없는 것 같습니다.

rgommers 에 2019년 08월 24일

이 페이지가 도움이 되었나요?

0 / 5 - 0 등급

Numpy: 배열과 같은 객체에 대해`np.vdot`에서 충돌이 발생합니다.

재현 코드 예 :

에러 메시지:

Numpy / Python 버전 정보 :

모든 6 댓글

관련 문제