o –hÆã@s(dZddlZddlZdd„Zdd„ZdS)z Utility methods to check that all processors are on the expected device Refactored since it can be used for multiple pipelines éNcCs€tj ¡s t d¡dS|j ¡D](\}}|jdur$t|jj   ¡ƒj }nt|j   ¡ƒj }t |ƒ d¡s9Jd|ƒ‚q|dƒdS)zU Check that the processors are all on the GPU and that basic execution works z\Unable to run the test that checks the pipeline is on the GPU, as there is no GPU available!NÚcudazProcessor %s was not on the GPUúThis is a small test)ÚtorchrÚ is_availableÚwarningsÚwarnÚ processorsÚitemsÚtrainerÚnextÚmodelÚ parametersÚdeviceÚ_modelÚstrÚ startswith©ÚpipelineÚnameÚprocr©rúf/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/pipeline/pipeline_device_tests.pyÚ check_on_gpu s    rcCsh|j ¡D](\}}|jdurt|jj ¡ƒj}nt|j ¡ƒj}t|ƒ  d¡s-Jd|ƒ‚q|dƒdS)zU Check that the processors are all on the CPU and that basic execution works NÚcpuzProcessor %s was not on the CPUr) rr r r r r rrrrrrrrÚ check_on_cpus   r)Ú__doc__rrrrrrrrÚs