ExecutionCPU

class nvmath.linalg.ExecutionCPU(*, num_threads: int | None = None)

A data class for providing CPU execution options.

num_threads

The number of CPU threads used to execute the operation. If not specified, defaults to the number of CPU cores available to the process.

Type:

int | None

See also

ExecutionCUDA, Matmul, matmul(), set_num_threads_local()