pytaco.get_num_threads

pytaco.get_num_threads()

Get the number of threads taco uses for computation.

Taco defaults to using one thread to perform computations.

The number of threads can be increased using set_num_threads() to perform computations in parallel. The user is encouraged to tune the number of threads in order to increase performance.

Returns

number_of_threads: int

The number of threads taco uses for performing tensor computations.

Examples

>>> import pytaco as pt
>>> pt.set_num_threads(1)
>>> pt.get_num_threads()
1