cerebras.modelzoo.config_manager.config_classes.base.run_config.CSTorchProfilerConfig#
- class cerebras.modelzoo.config_manager.config_classes.base.run_config.CSTorchProfilerConfig(host_activities: Optional[List[str]], start_step: int = <object object at 0x7fc286331b70>, end_step: int = <object object at 0x7fc286331b70>)[source]#
- host_activities: Optional[List[str]]#
list of act host, wgt host, csx host in the form of strings which should like, ACTHOST0, WGTHOST0, CSX0, etc
- start_step: int = <object object>#
step where to begin profiling
- end_step: int = <object object>#
step where to end profiling