4697 static int gaudi2_set_tpc_engine_mode(struct hl_device *hdev, u32 engine_id, u32 engine_command)  in gaudi2_set_tpc_engine_mode()  argument
4712 			(engine_command == HL_ENGINE_STALL) ? 1 : 0);  in gaudi2_set_tpc_engine_mode()
4715 	if (engine_command == HL_ENGINE_RESUME) {  in gaudi2_set_tpc_engine_mode()
4724 static int gaudi2_set_mme_engine_mode(struct hl_device *hdev, u32 engine_id, u32 engine_command)  in gaudi2_set_mme_engine_mode()  argument
4736 			(engine_command == HL_ENGINE_STALL) ? 1 : 0);  in gaudi2_set_mme_engine_mode()
4742 static int gaudi2_set_edma_engine_mode(struct hl_device *hdev, u32 engine_id, u32 engine_command)  in gaudi2_set_edma_engine_mode()  argument
4757 			(engine_command == HL_ENGINE_STALL) ? 1 : 0);  in gaudi2_set_edma_engine_mode()
4760 	if (engine_command == HL_ENGINE_STALL) {  in gaudi2_set_edma_engine_mode()
4770 		u32 *engine_ids, u32 num_engines, u32 engine_command)  in gaudi2_set_engine_modes()  argument
4780 			rc = gaudi2_set_tpc_engine_mode(hdev, engine_ids[i], engine_command);  in gaudi2_set_engine_modes()
4789 			rc = gaudi2_set_mme_engine_mode(hdev, engine_ids[i], engine_command);  in gaudi2_set_engine_modes()
4798 			rc = gaudi2_set_edma_engine_mode(hdev, engine_ids[i], engine_command);  in gaudi2_set_engine_modes()
4813 					u32 num_engines, u32 engine_command)  in gaudi2_set_engines()  argument
4815 	switch (engine_command) {  in gaudi2_set_engines()
4818 		return gaudi2_set_engine_cores(hdev, engine_ids, num_engines, engine_command);  in gaudi2_set_engines()
4822 		return gaudi2_set_engine_modes(hdev, engine_ids, num_engines, engine_command);  in gaudi2_set_engines()
4825 		dev_err(hdev->dev, "failed to execute command id %u\n", engine_command);  in gaudi2_set_engines()