mirror of
https://github.com/pybind/pybind11.git
synced 2026-03-14 20:27:47 +00:00
8ecf10e8cc
* Add a test reproducing the #5827 crash Signed-off-by: Rostan Tabet <rtabet@nvidia.com> * Fix #5827 Signed-off-by: Rostan Tabet <rtabet@nvidia.com> * Rename PYBIND11_HAS_BARRIER and move it to common.h Signed-off-by: Rostan Tabet <rtabet@nvidia.com> * In test_thread.{cpp,py}, rename has_barrier Signed-off-by: Rostan Tabet <rtabet@nvidia.com> --------- Signed-off-by: Rostan Tabet <rtabet@nvidia.com>
81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
import env
|
|
from pybind11_tests import thread as m
|
|
|
|
|
|
class Thread(threading.Thread):
|
|
def __init__(self, fn):
|
|
super().__init__()
|
|
self.fn = fn
|
|
self.e = None
|
|
|
|
def run(self):
|
|
try:
|
|
for i in range(10):
|
|
self.fn(i, i)
|
|
except Exception as e:
|
|
self.e = e
|
|
|
|
def join(self):
|
|
super().join()
|
|
if self.e:
|
|
raise self.e
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform.startswith("emscripten"), reason="Requires threads")
|
|
def test_implicit_conversion():
|
|
a = Thread(m.test)
|
|
b = Thread(m.test)
|
|
c = Thread(m.test)
|
|
for x in [a, b, c]:
|
|
x.start()
|
|
for x in [c, b, a]:
|
|
x.join()
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform.startswith("emscripten"), reason="Requires threads")
|
|
def test_implicit_conversion_no_gil():
|
|
a = Thread(m.test_no_gil)
|
|
b = Thread(m.test_no_gil)
|
|
c = Thread(m.test_no_gil)
|
|
for x in [a, b, c]:
|
|
x.start()
|
|
for x in [c, b, a]:
|
|
x.join()
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform.startswith("emscripten"), reason="Requires threads")
|
|
def test_bind_shared_instance():
|
|
nb_threads = 4
|
|
b = threading.Barrier(nb_threads)
|
|
|
|
def access_shared_instance():
|
|
b.wait()
|
|
for _ in range(1000):
|
|
m.EmptyStruct.SharedInstance # noqa: B018
|
|
|
|
threads = [
|
|
threading.Thread(target=access_shared_instance) for _ in range(nb_threads)
|
|
]
|
|
for thread in threads:
|
|
thread.start()
|
|
for thread in threads:
|
|
thread.join()
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform.startswith("emscripten"), reason="Requires threads")
|
|
@pytest.mark.skipif(not m.defined_PYBIND11_HAS_STD_BARRIER, reason="no <barrier>")
|
|
@pytest.mark.skipif(env.sys_is_gil_enabled(), reason="Deadlock with the GIL")
|
|
def test_pythread_state_clear_destructor():
|
|
class Foo:
|
|
def __del__(self):
|
|
m.acquire_gil()
|
|
|
|
m.test_pythread_state_clear_destructor(Foo)
|