3j JSrSSKrSSKrSSKrSSKJr SSKJrJr SSK J r Sr S\R"S5l S\R"S 5l S rS \\4S jrS rS\\S \\\44Sjr\"SS9S \\4Sj5rS\S \4SjrSS \\ R,4Sjjrg)zX Common utils for testing. These functions allow testing only some frameworks, not all. N) lru_cache)ListTuple) _backendszAlex RogozhnikovT tensorflow matplotlib)minmaxsummeanprodreturnc /n[RR5nU(a<UR5nXR5- nUR U5 U(aM<UVs/sHo3R PM sn$s snf)N)rAbstractBackend__subclasses__popappendframework_name)backend_subclassesbackendsbackendbs O/home/wildlama/miniconda3/lib/python3.13/site-packages/einops/tests/__init__.pyfind_names_of_all_frameworksrsp((779H ,,.**,,!!'* ('9 9&8  &8 99 9s(BEINOPS_TEST_BACKENDS backend_namesc|[5nUHnX!;dM [SU35e [SRU54$)NUnknown framework: ,)r RuntimeError ENVVAR_NAMEjoin)r_known_backends backend_names runparse_backendsr%%sC24O%  .!4\NCD D& / //)maxsizec[[R;a[S[S35e[R[R S5n[ 5nUHnX!;dM [SU35e U$)Nz/Testing frameworks were not specified, env var z not setrr)r!osenvironr splitr)parsed_backendsr#r$s rparse_backends_to_testr.-sp"**$L[MYabccjj-33C8O24O'  .!4\NCD D( r&rcTU[5;a[SU35eU[5;$)z8Used to skip test if corresponding backend is not testedzUnknown framework )rr r.)rs ris_backend_testedr0:s0244/y9:: ,. ..r&c|U(dU(dl[R[R[R[R[R [R [R/nOZ[R[R [R /nO*U(d[R/nO[R/n[5n/nUH,nURU;aMURU"55 M. U$![a [R"SU3SS9 MWf=f)z :param symbolic: symbolic or imperative frameworks? :param layers: layers or operations? :return: list of backends satisfying set conditions z,backend could not be initialized for tests: r') stacklevel)r NumpyBackend JaxBackend TorchBackendTensorflowBackendOneFlowBackend PaddleBackend CupyBackendPyTensorBackendTFKerasBackendr.rr ImportErrorwarningswarn)symboliclayers backend_typesbackend_names_to_testresult backend_types rcollect_test_backendsrEAs &&$$&&++((''%%M&&((''M ))M ((M34 F%  & &.C C  g MM,. ) & M  g MMHWde f gs9D"D;:D;)FF)__doc__loggingr*r= functoolsrtypingrreinopsr __author__ getLoggerdisabledFLOAT_REDUCTIONSstrrr!r%r.boolr0rrEr&rrRs    ,0,(+/,(8:d3i:% 0DI0%S/0 1 S  /s/t/,4 @Y@Y;Z,r&