Lines Matching refs:elem_size

26   CeedInt                  num_elem, num_comp, elem_size, comp_stride;  in CeedElemRestrictionSetupCompile_Hip()  local
37 CeedCallBackend(CeedElemRestrictionGetMaxPointsInElement(rstr, &elem_size)); in CeedElemRestrictionSetupCompile_Hip()
39 CeedCallBackend(CeedElemRestrictionGetElementSize(rstr, &elem_size)); in CeedElemRestrictionSetupCompile_Hip()
48 CeedInt strides[3] = {1, num_elem * elem_size, elem_size}; in CeedElemRestrictionSetupCompile_Hip()
54 …ip(ceed, restriction_kernel_source, &impl->module, 6, "RSTR_ELEM_SIZE", elem_size, "RSTR_NUM_ELEM"… in CeedElemRestrictionSetupCompile_Hip()
63 …ip(ceed, restriction_kernel_source, &impl->module, 6, "RSTR_ELEM_SIZE", elem_size, "RSTR_NUM_ELEM"… in CeedElemRestrictionSetupCompile_Hip()
74 …ip(ceed, restriction_kernel_source, &impl->module, 6, "RSTR_ELEM_SIZE", elem_size, "RSTR_NUM_ELEM"… in CeedElemRestrictionSetupCompile_Hip()
85 …ip(ceed, restriction_kernel_source, &impl->module, 6, "RSTR_ELEM_SIZE", elem_size, "RSTR_NUM_ELEM"… in CeedElemRestrictionSetupCompile_Hip()
98 …ip(ceed, restriction_kernel_source, &impl->module, 6, "RSTR_ELEM_SIZE", elem_size, "RSTR_NUM_ELEM"… in CeedElemRestrictionSetupCompile_Hip()
147 CeedInt elem_size; in CeedElemRestrictionApply_Hip_Core() local
149 CeedCallBackend(CeedElemRestrictionGetElementSize(rstr, &elem_size)); in CeedElemRestrictionApply_Hip_Core()
150 const CeedInt block_size = elem_size < 256 ? (elem_size > 64 ? elem_size : 64) : 256; in CeedElemRestrictionApply_Hip_Core()
417 static int CeedElemRestrictionOffset_Hip(const CeedElemRestriction rstr, const CeedInt elem_size, c… in CeedElemRestrictionOffset_Hip() argument
432 const CeedInt size_indices = num_elem * elem_size; in CeedElemRestrictionOffset_Hip()
459 …for (CeedInt i = 0; i < elem_size; ++i) ++t_offsets[ind_to_offset[indices[elem_size * e + i]] + 1]; in CeedElemRestrictionOffset_Hip()
465 for (CeedInt i = 0; i < elem_size; ++i) { in CeedElemRestrictionOffset_Hip()
466 const CeedInt lid = elem_size * e + i; in CeedElemRestrictionOffset_Hip()
503 CeedInt num_elem, num_comp, elem_size; in CeedElemRestrictionCreate_Hip() local
513 CeedCallBackend(CeedElemRestrictionGetElementSize(rstr, &elem_size)); in CeedElemRestrictionCreate_Hip()
522 elem_size = max_points; in CeedElemRestrictionCreate_Hip()
524 const CeedInt size = num_elem * elem_size; in CeedElemRestrictionCreate_Hip()
533 CeedInt layout[3] = {1, size, elem_size}; in CeedElemRestrictionCreate_Hip()
546 CeedSize offsets_len = elem_size * num_elem, at_points_size = num_elem + 1; in CeedElemRestrictionCreate_Hip()
547 CeedInt max_points = elem_size, *offsets_padded, *points_per_elem; in CeedElemRestrictionCreate_Hip()
578 … CeedCallBackend(CeedElemRestrictionSetAtPointsEVectorSize(rstr, elem_size * num_elem * num_comp)); in CeedElemRestrictionCreate_Hip()
597 … if (is_deterministic) CeedCallBackend(CeedElemRestrictionOffset_Hip(rstr, elem_size, offsets)); in CeedElemRestrictionCreate_Hip()
605 … if (is_deterministic) CeedCallBackend(CeedElemRestrictionOffset_Hip(rstr, elem_size, offsets)); in CeedElemRestrictionCreate_Hip()