243 static u32 test_states(unsigned int *tasks, u32 state_mask)  in test_states()  argument
245 	const bool oncpu = state_mask & PSI_ONCPU;  in test_states()
248 		state_mask |= BIT(PSI_IO_SOME);  in test_states()
250 			state_mask |= BIT(PSI_IO_FULL);  in test_states()
254 		state_mask |= BIT(PSI_MEM_SOME);  in test_states()
256 			state_mask |= BIT(PSI_MEM_FULL);  in test_states()
260 		state_mask |= BIT(PSI_CPU_SOME);  in test_states()
263 		state_mask |= BIT(PSI_CPU_FULL);  in test_states()
266 		state_mask |= BIT(PSI_NONIDLE);  in test_states()
268 	return state_mask;  in test_states()
281 	u32 state_mask;  in get_recent_times()  local
290 		state_mask = groupc->state_mask;  in get_recent_times()
308 		if (state_mask & (1 << s))  in get_recent_times()
771 	if (groupc->state_mask & (1 << PSI_IO_SOME)) {  in record_times()
773 		if (groupc->state_mask & (1 << PSI_IO_FULL))  in record_times()
777 	if (groupc->state_mask & (1 << PSI_MEM_SOME)) {  in record_times()
779 		if (groupc->state_mask & (1 << PSI_MEM_FULL))  in record_times()
783 	if (groupc->state_mask & (1 << PSI_CPU_SOME)) {  in record_times()
785 		if (groupc->state_mask & (1 << PSI_CPU_FULL))  in record_times()
789 	if (groupc->state_mask & (1 << PSI_NONIDLE))  in record_times()
802 	u32 state_mask;  in psi_group_change()  local
814 		state_mask = 0;  in psi_group_change()
817 		state_mask = PSI_ONCPU;  in psi_group_change()
820 		state_mask = groupc->state_mask & PSI_ONCPU;  in psi_group_change()
853 		if (unlikely(groupc->state_mask & (1 << PSI_NONIDLE)))  in psi_group_change()
856 		groupc->state_mask = state_mask;  in psi_group_change()
861 	state_mask = test_states(groupc->tasks, state_mask);  in psi_group_change()
871 	if (unlikely((state_mask & PSI_ONCPU) && cpu_curr(cpu)->in_memstall))  in psi_group_change()
872 		state_mask |= (1 << PSI_MEM_FULL);  in psi_group_change()
876 	groupc->state_mask = state_mask;  in psi_group_change()
878 	if (state_mask & group->rtpoll_states)  in psi_group_change()
946 			if (groupc->state_mask & PSI_ONCPU) {  in psi_task_switch()