Module neuralops.config.hardware_config

Classes

class HardwareConfig (device: str = 'auto',
force_cpu_for_svd: bool = True,
enable_fp16: bool = False,
enable_bf16: bool = False,
num_workers: int = 4,
pin_memory: bool = False,
torch_compile: bool = False,
empty_cache_every_step: bool = False)

Expand source code

@dataclass
class HardwareConfig:
    device: str = "auto"  # auto, cuda, mps, cpu
    force_cpu_for_svd: bool = True
    enable_fp16: bool = False
    enable_bf16: bool = False
    num_workers: int = 4
    pin_memory: bool = False
    torch_compile: bool = False
    empty_cache_every_step: bool = False

HardwareConfig(device: str = 'auto', force_cpu_for_svd: bool = True, enable_fp16: bool = False, enable_bf16: bool = False, num_workers: int = 4, pin_memory: bool = False, torch_compile: bool = False, empty_cache_every_step: bool = False)

Instance variables

var device : str

The type of the None singleton.

var empty_cache_every_step : bool

The type of the None singleton.

var enable_bf16 : bool

The type of the None singleton.

var enable_fp16 : bool

The type of the None singleton.

var force_cpu_for_svd : bool

The type of the None singleton.

var num_workers : int

The type of the None singleton.

var pin_memory : bool

The type of the None singleton.

var torch_compile : bool

The type of the None singleton.