blob: d6b0167d9ea9f40da2e5dd50833568ac659f6f24 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
# tag: openmp
from cython.parallel import parallel
from cython.cimports.openmp import omp_get_thread_num
import cython
@cython.cfunc
@cython.nogil
def long_running_task1() -> cython.void:
pass
@cython.cfunc
@cython.nogil
def long_running_task2() -> cython.void:
pass
def do_two_tasks():
thread_num: cython.int
with cython.nogil, parallel(num_threads=2):
thread_num = omp_get_thread_num()
if thread_num == 0:
long_running_task1()
elif thread_num == 1:
long_running_task2()
|
