9j:I`SSKJr SSKrSSKrSSKJr SSK J r Sr Sr SrSrS rS S jrg) ) annotationsN)cublas)devicec SSKJn URS:wa$[SR UR55eURS;a$[SR UR55eUR SUR S:wa [S5eUR SUR S:wa/[S R UR UR 55eUR UR :wa/[S R UR UR 55eUR nUS :XaS nO?US :XaS nO6US:XaSnO-US:XaSnO$[SR UR 55e[X$S-5n[X$S-5n[X$S-5nUR SnURS:XaUR SOSn UR(a[Rn O-UR(a[Rn O [S5eUR(d [S5e[R"5n [ R""U[$R&S9n [ R""S[$R&S9n U"XXR(R*U5n[ R""XR S9nU"XXR(R*XR(R*U R(R*U R(R*5 [ R,R.R1Xm5 U"XXUR(R*UU R(R*UR(R*XR(R*5 [ R,R.R1X}5 g)aSolve a linear matrix equation using cusolverDngetr[fs](). Computes the solution to a system of linear equation ``ax = b``. Args: a (cupy.ndarray): The matrix with dimension ``(M, M)``. b (cupy.ndarray): The matrix with dimension ``(M)`` or ``(M, K)``. Returns: cupy.ndarray: The matrix with dimension ``(M)`` or ``(M, K)``. Note: ``a`` and ``b`` will be overwritten. rcusolvera.ndim must be 2 (actual: {}))r "b.ndim must be 1 or 2 (actual: {})r za must be a square matrix.shape mismatch (a: {}, b: {}).zdtype mismatch (a: {}, b: {})fsdFcDzzunsupported dtype (actual:{})getrf_bufferSizegetrfgetrsz'a must be F-contiguous or C-contiguous.zb must be F-contiguous.dtypeN)cupy_backends.cuda.libsrndim ValueErrorformatshaper TypeErrorgetattr _f_contiguous_cublas CUBLAS_OP_N _c_contiguous CUBLAS_OP_T_deviceget_cusolver_handle_cupyempty_numpyint32dataptrlinalg_util3_check_cusolver_dev_info_if_synchronization_allowed)ab _cusolverrthelperrrnnrhstranshandledipivdinfolworkdworks F/home/wildlama/miniconda3/lib/python3.13/site-packages/cupyx/lapack.pygesvr? s>vv{8??GHHvvV=DDQVVLMMwwqzQWWQZ566wwqzQWWQZ913 3ww!''7qww02 2 GGE |  #  #  # 7>>qwwGHH Y$6 6 7F I7{ +E I7{ +E  A11771:!D## ##BCC ??233  ( ( *F KK .E KK .E 6aQ /E KKWW -E &Q Azz~~uzz~~ **.. LLJJ  &!&&**a **..!&&**a9 LLJJ cSSKJn URS:wa$[SR UR55eURS:XaSnODURS:XaUR SnO$[SR UR55eUR SUR S:wa/[SR UR UR 55eUR UR :wa/[SR UR UR 55eUR nUS :XaS nO5US :XaS nO,US :XaS nO#US:XaSnO[SR U55e[X%S-5n[X%S-5n[[US-5nUS;a[X%S-5n [X%S-5n O[X%S-5n [X%S-5n [Rn URS;a[Rn O[Rn UR up[X5n[R"S[ R"S9n[R"XS9n[$R&"5n[$R("5n[ R*"SUS9nX:GauUR-S S9nUR-S S9nU"UXUR.R0U 5n[R"UUS9nU"UXUR.R0U UR.R0UR.R0UUR.R05 [R2R4R7UU5 U "U[R8XX?UR.R0U UR.R0UR.R0U 5 n[R"UUS9nU "U[R8XUXR.R0U UR.R0UR.R0U UR.R0UUR.R05 [R2R4R7U U5 U"U[R8[R:U [R<XUR>R.UR.R0XR.R0U 5 USU$URA5RBR-S S9nUnURS:XaU4OX4n[RD"UUS S9nUUSU &U"UXUR.R0U5n[R"UUS9nU"UXUR.R0UUR.R0UR.R0UUR.R05 [R2R4R7UU5 U"U[R8[R:U [R<XUR>R.UR.R0XR.R0U5 U "U[R8XUXR.R0UUR.R0UR.R0U5 n[R"UUS9nU "U[R8XUXR.R0UUR.R0UR.R0UUR.R0UUR.R05 [R2R4R7U U5 U$)awSolves over/well/under-determined linear systems. Computes least-square solution to equation ``ax = b` by QR factorization using cusolverDngeqrf(). Args: a (cupy.ndarray): The matrix with dimension ``(M, N)``. b (cupy.ndarray): The matrix with dimension ``(M)`` or ``(M, K)``. Returns: cupy.ndarray: The matrix with dimension ``(N)`` or ``(N, K)``. rrr r r r r zdtype mismatch (a: {}, b: {}).rrrrrrrzunsupported dtype (actual: {})geqrf_bufferSizegeqrftrsmsdormqr_bufferSizeormqrunmqr_bufferSizeunmqrfdrg?)orderNrrK)#rrrrrrrr r"r#charr% CUBLAS_OP_Cminr(r)r*r+r&r'get_cublas_handlearraycopyr,r-r.r/r0CUBLAS_SIDE_LEFTCUBLAS_FILL_MODE_UPPERCUBLAS_DIAG_NON_UNITctypesconjTzeros)r1r2r3r7rr4 geqrf_helperrCrD ormqr_helperrGno_transr8mr6mn_mindev_infotaucusolver_handle cublas_handleonews_size workspacebb out_shapes r>gelsrhRs>vv{8??GHHvv{ 1wwqz=DDQVVLMMwwqzQWWQZ913 3ww!''913 3 GGE |  #  #  # 9@@GHH9*<&<=L I7{ +E 7AJ 'DDyy.@*@A  w;/y.@*@A  w;/""H zzT#### 77DA YF{{1FLL1H ++f *C113O--/M ,,s% (Cv FFF  FFF affjj!DKKu5  oqQVVZZCHHLLnn  '8==+<+< > NN 8  W55u FFJJ388<<Q8KKu5  ow774ffjj!SXX\\166::qnn  '8==+<+< >  NN 8  ]G44  + +X  ) )6 ZZ__affjj!VVZZ < !u  FFHJJOO#O & FFaKQDaY KK c :"1affjj!DKKu5  oqQVVZZCHHLLnn  '8==+<+< > NN 8  ]G44  + +U  ) )1 ZZ__affjj!VVZZ <  W55xD FFJJ388<<Q@KKu5  ow77dffjj!SXX\\166::qnn  '8==+<+< >  NN 8 r@c SSKJn SSKnURRS5(d [ S5e[ R "URUR5n[ R "US5nUS:XaURnURnOUS:XaURnURnOdUS:XaURnURnOEUS:XaURnURnO&S R!UR5n[#U5eUR%US S S 9n[&R(R+U5nUR,S Sup[/[ R0"UR,SS 55n [2R4"5n [6R8n [&R:"U [ R<S9nU"XXR>R@XR>R@U 5 [&RBRDRGX^5 UR,nURI5RKXS5R%US S S 9n[&R(R+U5nUR,S Sunn[&R:"S[ R<S9nU"XU UUR>R@U UR>R@UUR>R@U 5 [&RBRDRGXn5 URI5RKU5$)Nrr potrsBatchedpotrsBatched is not availablerrrrDdtype must be float32, float64, complex64 or complex128 (actual: {})CTrKrRrr )&rrcupyx.cusolvercheck_availability RuntimeErrorr* promote_typesr spotrfBatched spotrsBatched dpotrfBatched dpotrsBatched cpotrfBatched cpotrsBatched zpotrfBatched zpotrsBatchedrrastyper(_core _mat_ptrsrintprodr&r'r"CUBLAS_FILL_MODE_LOWERr)r+r,r-r.r/r0rWreshape)r1r2r3cupyxr potrfBatchedrjmsgapldar6 batch_sizer9uplor_b_shapebpldbr7s r> _batched_posvrs= >> , ,^ < <:;;  !'' 2E   ,E | ..  .. # ..  .. # ..  .. # ..  .. %vagg o c-A   q !B WWRS\FCV[["./J  ( ( *F  ) )D{{:V\\:Hq''++sMM4E4E LLJJ ggG +225$2OA   q !B IC{{1FLL1Hq$ S"''++s""J0 LLJJ  668  G $$r@c SSKJn [RRnUR X5 UR U5 URU5 URS:a [X5$[R"URUR5n[R"US5nUS:Xa%URnURnURnOUS:Xa%UR nUR"nUR$nO|US:Xa%UR&nUR(nUR*nOQUS:Xa%UR,nUR.nUR0nO&SR3UR5n[5U5eUR7USS S 9nUR8up[:R<"5n [>R@n [RB"S [RDS 9n U"XXRFRHU 5n[RB"XS 9nU"XXRFRHXRFRHXRFRH5 [RRRKX]5 UR8nURMU S 5R7USS S 9nUR8unnU"XU UURFRHXRFRHUU RFRH5 [RRRKX}5 [RN"URMU55$)a$Solve the linear equations A x = b via Cholesky factorization of A, where A is a real symmetric or complex Hermitian positive-definite matrix. If matrix ``A`` is not positive definite, Cholesky factorization fails and it raises an error. Note: For batch input, NRHS > 1 is not currently supported. Args: a (cupy.ndarray): Array of real symmetric or complex hermitian matrices with dimension (..., N, N). b (cupy.ndarray): right-hand side (..., N) or (..., N, NRHS). Returns: x (cupy.ndarray): The solution (shape matches b). rrr rrrrrlTrnr rrp)(rrr(r.r/_assert_cupy_array_assert_stacked_2d_assert_stacked_squarerrr*rtrspotrfspotrf_bufferSizespotrsdpotrfdpotrf_bufferSizedpotrscpotrfcpotrf_bufferSizecpotrszpotrfzpotrf_bufferSizezpotrsrrr}rr&r'r"rr)r+r,r-r0rascontiguousarray)r1r2r3r/rpotrfpotrf_bufferSizepotrsrrr6r9rr_worksizererrr7s r>posvrs > LL  E Q" Q   #vvzQ""  !'' 2E   ,E |  $66   #  $66   #  $66   #  $66  %vagg o c-A WWFC  ( ( *F  ) )D{{1FLL1HaSAH H2I &66::sNN,>,> MM%%' LLJJ ggG !RStR@O[>RBURSURSURDRFURSURDRFURSU RDRF5 [RRRIX5 URR"(a*UR&UR&:aUR)5nURKU5$)aImplements lapack XPOTRS through cusolver.potrs. Solves linear system A x = b given the cholesky decomposition of A, namely L. Supports also batches of linear systems and more than one right-hand side (NRHS > 1). Args: L (cupy.ndarray): Array of Cholesky decomposition of real symmetric or complex hermitian matrices with dimension (..., N, N). b (cupy.ndarray): right-hand side (..., N) or (..., N, NRHS). Note that this array may be modified in place, as usually done in LAPACK. lower (bool): If True, L is lower triangular. If False, L is upper triangular. Returns: cupy.ndarray: The solution to the linear system. Note this may point to the same memory as b, since b may be modified in place. .. warning:: This function calls one or more cuSOLVER routine(s) which may yield invalid results if input conditions are not met. To detect these invalid results, you can set the `linalg` configuration to a value that is not `ignore` in :func:`cupyx.errstate` or :func:`cupyx.seterr`. rrr rpr Nzb is not a vector or a matrixz.length of arrays in b does not match size of LrF)rRrrrrlr)&rrr(r.r/rrrr_batched_potrsrr*rtrr}flags f_contiguous c_contiguousasfortranarraysizerWrrrrrrr&r'r)r+r"rrTr,r-r0r) Lr2lowerr3r/r6rrrrr9r_s r>rras4> LL  E Q" Q   # vvzaE**  AggGvv{ agJ 66Q;777; 771:?LLL?  !'' 2E   ,E 88E8 &U(Cq GG 177+?+?   #ww  66AFF?AA 5!''"6"6   # |   #   #   #  %vagg o  ( ( *F{{1EKK0H */&&&&               LLJJ  ww FFH 99W r@c SSKJn SSKnURRS5(d [ S5eUR UR S- :dS5eUR nUR UR :aUSnUR SSUR SS:XdS 5e[R"URUR5n[R"US 5nURUS S S 9nURUS S S 9nURR(aURR(dS5eU(+nUS :Xa URnO_US:Xa URnOLUS:Xa URnO9US:Xa UR nO&SR#UR5n[%U5e[&R("5n [*R,"S[R.S9n [R0"UR SS5n UR Sn UR35n[*R4R7U5n UR SnUS:XaUSnO-[*R,"UR SSURS S9n[*R4R7US5n[9UR S5HnUS:a USU4US'U"U U(a[:R<O[:R>U SU R@RBUR SUR@RBUR SU R@RBU 5 [*RDRFRIXz5 US:aUR35USU4'MUR35nM URKU5$)Nrrrjrkr z8Batch dimension of b is different than that of L).Nroz'Batch dimension of L and b do not matchrrmFrnz"Unexpected non C-contiguous arraysrrrrlrrp).rrL.)&rrrqrrrsrrr*rtrr}rrrvrxrzr|rrr&r'r(r)r+rrWr~rranger"rrTr,r-r.r/r0r)rr2rr3rrrrjrr9r_rr6L_pr7b_tmpb_tmp_pis r>rrs8= >> , ,^ < <:;; 66QVVAX   ggGvv iL 773B<1773B< '212 '  !'' 2E   ,E c.A c.A 77  AGG$8$8-,- 8IE | .. # .. # .. # .. %vagg o ( ( *F{{1FLL1HQWWSb\*J  A A ++   "C 772;D qy&  AGGCRLsCkk##E)$45G1772;  !836E#J .3G * *  * * HHLL GGBK LL   KKO MM      NN  $ !8 Ac1fI A1 6 99W r@)rbool) __future__rnumpyr*cupyr(rrr" cupy.cudarr&r?rhrrrrr@r>rs<"5'EPBJ8%vL7^_DSr@