nncf.torch

Base subpackage for NNCF PyTorch functionality.

Functions

disable_tracing(method)	Patch a method so that it will be executed within no_nncf_trace context
force_build_cpu_extensions()
force_build_cuda_extensions()

nncf.torch.disable_tracing(method)

Patch a method so that it will be executed within no_nncf_trace context :param method: A method to patch.

Parameters:

method (Callable[Ellipsis, Any]) –

Return type:

None

nncf.torch.force_build_cpu_extensions()

nncf.torch.force_build_cuda_extensions()