Sh dZddlmZmZmZmZddlmZdgZ d dee de ded eeee dfd ee e ff d Z dS) z0Determines if a contraction can use BLAS or not.)ListSequenceTupleUnion)ArrayIndexTypecan_blasNinputsresult idx_removedshapesreturnct|dkrdS|\}}t||zD]b}||||}}|dks|dks ||zdkrdS||zdz t||vkrdSc|L|D]I}|d|||d||krdSJt|dkrdSd|D} | d|z } | d|z } t|} |d|dkrdS| d| dkrd S|| d|d| krd S|d| || dkrd S|| d|| dkrd S|d| |d| krd St| dkst| dkrd Sd S) 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/EINSUMc,g|]}t|S)set).0xs U/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/opt_einsum/blas.py zcan_blas..Ws # # #qCFF # # #DOTz DOT/EINSUMGEMMz GEMV/EINSUMTDOT)lenrcountintfind) r r r r input_left input_rightcnlnrsets keep_left keep_rightrss rrr sc^ 6{{au$J k) * *  !!!$$k&7&7&:&:B FFQBGaKK55 7Q;#a6k** * *55 +   Aay++,q +:J:J1:M:M0NNNuuO  ;1~ $ #F # # #DQ+%Ia;&J [  BayF1Iu aDG  |2#$$;ss+++v CRCK- - -v RCDD [". . .v CRCK, , ,v i..A  3z??a#7#7}vr)N) __doc__typingrrrropt_einsum.typingr__all__strrboolrrrrr/s66////////////,,,,,, ,15 pp Ip p p (5:&, - p  39 ppppppr