,Vh xdZddlmZmZmZmZddlmZdgZ d dee de ded eeee dfd ee e ff d Z y) z0Determines if a contraction can use BLAS or not.)ListSequenceTupleUnion)ArrayIndexTypecan_blasNinputsresult idx_removedshapesreturnct|dk7ry|\}}t||zD]P}|j||j|}}|dkDs |dkDs||zdkDry||zdz t||vk(sPy|8|D]3}|d|j ||d|j |k7s3yt|dk(ry|D cgc] } t| } } | d|z } | d|z } t|} |d|dk(ry| d| dk(ry|| d|d| k(ry |d| || dk(ry || d|| dk(ry |d| |d| k(ry t| dk(st| dk(ry y cc} w) aChecks if we can use a BLAS call. Parameters ---------- inputs : list of str Specifies the subscripts for summation. result : str Resulting summation. idx_removed : set Indices that are removed in the summation shapes : sequence of tuple[int], optional If given, check also that none of the indices are broadcast dimensions. Returns: ------- type : str or bool The type of BLAS call to be used or False if none. Notes: ----- We assume several operations are not efficient such as a transposed DDOT, therefore 'ijk,jki->' should prefer einsum. These return the blas type appended with "/EINSUM" to differentiate when they can still be done with tensordot if required, e.g. when a backend has no einsum. Examples: -------- >>> can_blas(['ij', 'jk'], 'ik', set('j')) 'GEMM' >>> can_blas(['ijj', 'jk'], 'ik', set('j')) False >>> can_blas(['ab', 'cd'], 'abcd', set()) 'OUTER/EINSUM' >>> # looks like GEMM but actually 'j' is broadcast: >>> can_blas(['ij', 'jk'], 'ik', set('j'), shapes=[(4, 1), (5, 6)]) False FNrz OUTER/EINSUMDOTz DOT/EINSUMGEMMz GEMV/EINSUMTDOT)lensetcountintfind)r r r r input_left input_rightcnlnrxsets keep_left keep_rightrss ?/home/dcms/DCMS/lib/python3.12/site-packages/opt_einsum/blas.pyrr s^ 6{a$J k) * !!!$k&7&7&:B FQBGaK 7Q;#a6k* *  Aay+,q +:J:J1:M0NN   ;1# #qCF #D #Q+%Ia;&J [ BayF1I aDG 2#$;s++ CRK- - RCD [". . CRK, , i.A 3z?a#7G $sE)N) __doc__typingrrrropt_einsum.typingr__all__strrboolrr#r,sv6//, ,15 p Ip p p (5:&, - p  39 pr+