63 				 unsigned int ae, unsigned int image_num)  in qat_uclo_init_ae_data()  argument
70 	ae_data = &obj_handle->ae_data[ae];  in qat_uclo_init_ae_data()
159 				      unsigned char ae, unsigned int addr,  in qat_uclo_wr_umem_by_words()  argument
170 		qat_hal_wr_umem(handle, ae, addr++, 1, &outval);  in qat_uclo_wr_umem_by_words()
177 				   unsigned char ae,  in qat_uclo_batch_wr_umem()  argument
189 		ae = umem_init->ae;  in qat_uclo_batch_wr_umem()
193 		qat_uclo_wr_umem_by_words(handle, ae, addr, value, size);  in qat_uclo_batch_wr_umem()
218 	unsigned long ae = 0;  in qat_uclo_parse_num()  local
228 	if ((kstrtoul(buf, 10, &ae)))  in qat_uclo_parse_num()
231 	*num = (unsigned int)ae;  in qat_uclo_parse_num()
237 				     unsigned int size_range, unsigned int *ae)  in qat_uclo_fetch_initmem_ae()  argument
255 	if (qat_uclo_parse_num(str, ae)) {  in qat_uclo_fetch_initmem_ae()
259 	if (*ae >= ICP_QAT_UCLO_MAX_AE) {  in qat_uclo_fetch_initmem_ae()
260 		pr_err("QAT: ae %d out of range\n", *ae);  in qat_uclo_fetch_initmem_ae()
268 					   *init_mem, unsigned int ae,  in qat_uclo_create_batch_init_list()  argument
298 		mem_init->ae = ae;  in qat_uclo_create_batch_init_list()
324 	unsigned int ae;  in qat_uclo_init_lmem_seg()  local
327 				      ICP_QAT_UCLO_MAX_LMEM_REG, &ae))  in qat_uclo_init_lmem_seg()
329 	if (qat_uclo_create_batch_init_list(handle, init_mem, ae,  in qat_uclo_init_lmem_seg()
330 					    &obj_handle->lm_init_tab[ae]))  in qat_uclo_init_lmem_seg()
339 	unsigned int ae, ustore_size, uaddr, i;  in qat_uclo_init_umem_seg()  local
342 	if (qat_uclo_fetch_initmem_ae(handle, init_mem, ustore_size, &ae))  in qat_uclo_init_umem_seg()
344 	if (qat_uclo_create_batch_init_list(handle, init_mem, ae,  in qat_uclo_init_umem_seg()
345 					    &obj_handle->umem_init_tab[ae]))  in qat_uclo_init_umem_seg()
349 	for (i = 0; i < obj_handle->ae_data[ae].slice_num; i++) {  in qat_uclo_init_umem_seg()
350 		if (obj_handle->ae_data[ae].ae_slices[i].  in qat_uclo_init_umem_seg()
352 			obj_handle->ae_data[ae].ae_slices[i].  in qat_uclo_init_umem_seg()
385 	unsigned char ae;  in qat_uclo_init_ustore()  local
401 	for (ae = 0; ae < handle->hal_handle->ae_max_num; ae++) {  in qat_uclo_init_ustore()
402 		if (!test_bit(ae, (unsigned long *)&uof_image->ae_assigned))  in qat_uclo_init_ustore()
404 		ustore_size = obj_handle->ae_data[ae].eff_ustore_size;  in qat_uclo_init_ustore()
407 		qat_hal_wr_uwords(handle, (unsigned char)ae, 0,  in qat_uclo_init_ustore()
409 		qat_hal_wr_uwords(handle, (unsigned char)ae, patt_pos,  in qat_uclo_init_ustore()
419 	int i, ae;  in qat_uclo_init_memory()  local
434 	for (ae = 0; ae < handle->hal_handle->ae_max_num; ae++) {  in qat_uclo_init_memory()
435 		if (qat_hal_batch_wr_lm(handle, ae,  in qat_uclo_init_memory()
436 					obj_handle->lm_init_tab[ae])) {  in qat_uclo_init_memory()
437 			pr_err("QAT: fail to batch init lmem for AE %d\n", ae);  in qat_uclo_init_memory()
441 						 &obj_handle->lm_init_tab[ae]);  in qat_uclo_init_memory()
442 		qat_uclo_batch_wr_umem(handle, ae,  in qat_uclo_init_memory()
443 				       obj_handle->umem_init_tab[ae]);  in qat_uclo_init_memory()
446 						 umem_init_tab[ae]);  in qat_uclo_init_memory()
662 	int i, ae;  in qat_uclo_map_ae()  local
666 	for (ae = 0; ae <= max_ae; ae++) {  in qat_uclo_map_ae()
667 		if (!test_bit(ae,  in qat_uclo_map_ae()
671 			if (!test_bit(ae, (unsigned long *)  in qat_uclo_map_ae()
675 			if (qat_uclo_init_ae_data(obj_handle, ae, i))  in qat_uclo_map_ae()
743 			     unsigned char ae, unsigned char ctx_mask,  in qat_uclo_init_reg()  argument
753 		return qat_hal_init_gpr(handle, ae, ctx_mask, reg_type,  in qat_uclo_init_reg()
764 		return qat_hal_init_rd_xfer(handle, ae, ctx_mask, reg_type,  in qat_uclo_init_reg()
771 		return qat_hal_init_wr_xfer(handle, ae, ctx_mask, reg_type,  in qat_uclo_init_reg()
774 		return qat_hal_init_nn(handle, ae, ctx_mask, reg_addr, value);  in qat_uclo_init_reg()
783 				 unsigned int ae,  in qat_uclo_init_reg_sym()  argument
803 			qat_uclo_init_reg(handle, ae, ctx_mask,  in qat_uclo_init_reg_sym()
816 			qat_uclo_init_reg(handle, ae,  in qat_uclo_init_reg_sym()
840 	unsigned int s, ae;  in qat_uclo_init_globals()  local
850 	for (ae = 0; ae < handle->hal_handle->ae_max_num; ae++) {  in qat_uclo_init_globals()
851 		for (s = 0; s < obj_handle->ae_data[ae].slice_num; s++) {  in qat_uclo_init_globals()
852 			if (!obj_handle->ae_data[ae].ae_slices[s].encap_image)  in qat_uclo_init_globals()
854 			if (qat_uclo_init_reg_sym(handle, ae,  in qat_uclo_init_globals()
855 						  obj_handle->ae_data[ae].  in qat_uclo_init_globals()
866 	unsigned char ae, nn_mode, s;  in qat_uclo_set_ae_mode()  local
871 	for (ae = 0; ae < handle->hal_handle->ae_max_num; ae++) {  in qat_uclo_set_ae_mode()
872 		if (!test_bit(ae,  in qat_uclo_set_ae_mode()
875 		ae_data = &obj_handle->ae_data[ae];  in qat_uclo_set_ae_mode()
878 			if (!obj_handle->ae_data[ae].ae_slices[s].encap_image)  in qat_uclo_set_ae_mode()
881 			if (qat_hal_set_ae_ctx_mode(handle, ae,  in qat_uclo_set_ae_mode()
888 			if (qat_hal_set_ae_nn_mode(handle, ae, nn_mode)) {  in qat_uclo_set_ae_mode()
892 			if (qat_hal_set_ae_lm_mode(handle, ae, ICP_LMEM0,  in qat_uclo_set_ae_mode()
898 			if (qat_hal_set_ae_lm_mode(handle, ae, ICP_LMEM1,  in qat_uclo_set_ae_mode()
925 	unsigned int ae;  in qat_uclo_parse_uof_obj()  local
965 	for (ae = 0; ae < obj_handle->uimage_num; ae++)  in qat_uclo_parse_uof_obj()
966 		kfree(obj_handle->ae_uimage[ae].page);  in qat_uclo_parse_uof_obj()
1071 					*encap_page, unsigned int ae)  in qat_uclo_wr_uimage_raw_page()  argument
1098 		qat_hal_wr_uwords(handle, (unsigned char)ae,  in qat_uclo_wr_uimage_raw_page()
1114 	unsigned char ae;  in qat_uclo_wr_uimage_page()  local
1123 	for (ae = 0; ae < handle->hal_handle->ae_max_num; ae++) {  in qat_uclo_wr_uimage_page()
1124 		if (!test_bit(ae, (unsigned long *)&image->ae_assigned))  in qat_uclo_wr_uimage_page()
1127 		for (s = 0; s < obj_handle->ae_data[ae].slice_num; s++) {  in qat_uclo_wr_uimage_page()
1128 			if (image->ctx_assigned & obj_handle->ae_data[ae].  in qat_uclo_wr_uimage_page()
1132 		if (s >= obj_handle->ae_data[ae].slice_num)  in qat_uclo_wr_uimage_page()
1134 		page = obj_handle->ae_data[ae].ae_slices[s].page;  in qat_uclo_wr_uimage_page()
1137 		qat_uclo_wr_uimage_raw_page(handle, page->encap_page, ae);  in qat_uclo_wr_uimage_page()
1139 		page = obj_handle->ae_data[ae].ae_slices[s].page;  in qat_uclo_wr_uimage_page()
1141 			obj_handle->ae_data[ae].ae_slices[s].cur_page[ctx] =  in qat_uclo_wr_uimage_page()
1143 		qat_hal_set_live_ctx(handle, (unsigned char)ae,  in qat_uclo_wr_uimage_page()
1145 		qat_hal_set_pc(handle, (unsigned char)ae, image->ctx_assigned,  in qat_uclo_wr_uimage_page()