U
‰dá ã @ s\ d dl mZ d dlmZmZ d dlZd dlZd dlmZ ej j
Z
dd„ Z dd„ Z d d
„ Z
dS ) é )Ú
caffe2_pb2)ÚcoreÚ workspaceN)ÚZipFilec
C sŽ t | ƒ\}}}t|ƒ}t|ƒ}t|ƒ}|| }ttj |d¡dƒ@} | d¡ | d¡ | d¡ | d ||t|| d ƒd ¡¡ | d ¡ | d
¡ | d |¡¡ t |ƒD ]}
| d |
d
¡ q´| d¡ | d¡ | d
¡ | d |¡¡ t |ƒD ]}
| d |
d
¡ q| d¡ | d¡ | d¡ | d
¡ | d |¡¡ t |ƒD ]}
| d |
d
¡ q\| d¡ W 5 Q R X d S )NzSerializedTestCoverage.mdzw+z"# Serialized Test Coverage Report
z©This is an automatically generated file. Please see `caffe2/python/serialized_test/README.md` for details. In the case of merge conflicts, please rebase and regenerate.
z ## Summary
z5Serialized tests have covered {}/{} ({}%) operators
iè é
z## Not covered operators
z
z7There are {} not covered operators
z* Ú
z
z## Covered operators
z3There are {} covered operators
z%## Excluded from coverage statistics
z### Schemaless operators
z6There are {} schemaless operators
)
Úgen_coverage_setsÚlenÚopenÚosÚpathÚjoinÚwriteÚformatÚintÚsorted) Ú
source_dirÚ
output_dirÚcoveredÚ not_coveredZ
schemalessZ num_coveredZnum_not_coveredZnum_schemalessZ total_opsÚfÚn© r úJ/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/serialized_test/coverage.pyÚ gen_serialized_test_coverage s^
þÿ
ÿÿ
ÿÿ
ÿÿ r c C st t | ƒ}tƒ }g }t ¡ D ]N}t |¡}|d k r8|jr8q|rP||krh| |¡ q| d¡dkr| |¡ q|||fS )NZ_ENGINE_éÿÿÿÿ)
Úgen_covered_opsÚsetr Z_GetRegisteredOperatorsÚOpSchemaÚgetÚprivateÚaddÚfindÚappend)r Z covered_opsZnot_covered_opsZschemaless_opsZop_nameÚsr r r r D s
r c
C s dd„ }t ƒ }t | ¡D ]ú}tj | |¡}tj |¡s8qt ¡ }t|ƒ}| |¡ W 5 Q R X tj |d¡}t
|dƒ}| ¡ }W 5 Q R X ||ƒ} | | j
¡ d}
tj |d |
¡¡} tj | ¡rt
| dƒ}| ¡ } W 5 Q R X || ƒ}
| |
j
¡ |
d7 }
tj |d |
¡¡} q¶q|S )Nc S s t ¡ }| | ¡ |S )N)r Z OperatorDefZParseFromString)ÚxÚprotor r r Ú parse_protoX s
z$gen_covered_ops..parse_protozop.pbÚrbr z
grad_{}.pbé )r r Úlistdirr r
ÚisfileÚtempfileÚmkdtempr Ú
extractallr
Úreadr! Útyper )r r' r r ÚzipfileÚtemp_dirÚzZop_pathZ loaded_opZop_protoÚindexZ grad_pathZ loaded_gradZ
grad_protor r r r W s0
r )Z caffe2.protor Z
caffe2.pythonr r r r, r1 r ÚCr r r r r r r r Ú s /