File: manual_work.py

package info (click to toggle)
cython 3.1.6%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,932 kB
  • sloc: python: 92,172; ansic: 19,275; cpp: 1,407; xml: 1,031; javascript: 511; makefile: 373; sh: 223; sed: 11
file content (24 lines) | stat: -rw-r--r-- 540 bytes parent folder | download | duplicates (2)
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()