Skip to main content

Mojo function

form_q

form_q[dtype: DType, element_layout: Layout](sigma: LayoutTensor[dtype, sigma.layout, sigma.origin, address_space=sigma.address_space, element_layout=element_layout, layout_int_type=sigma.layout_int_type, linear_idx_type=sigma.linear_idx_type, masked=sigma.masked, alignment=sigma.alignment], A: LayoutTensor[dtype, A.layout, A.origin, address_space=A.address_space, element_layout=element_layout, layout_int_type=A.layout_int_type, linear_idx_type=A.linear_idx_type, masked=A.masked, alignment=A.alignment], Q: LayoutTensor[dtype, Q.layout, Q.origin, address_space=Q.address_space, element_layout=element_layout, layout_int_type=Q.layout_int_type, linear_idx_type=Q.linear_idx_type, masked=Q.masked, alignment=Q.alignment])

Forms the Q factor from the implicit Q factor stored in A and sigma after calling qr_factorization and stores the result in Q.

Was this page helpful?