o -i@sRddlZddlZddlmZddlmZddlmZejGdddZ e Z dS)N) Generator)contextmanager)Anyc@seZdZUdZeed<dZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <dZ eed <dZeed <dZeed <dZeed <dddZedededfddZdS)CompilationCounterrnum_models_seennum_graphs_seennum_piecewise_graphs_seen$num_piecewise_capturable_graphs_seennum_backend_compilationsnum_gpu_runner_capture_triggersnum_cudagraph_capturednum_inductor_compilesnum_eager_compilesnum_cache_entries_updatednum_compiled_artifacts_savedstock_torch_compile_countreturncCs t|S)N)copydeepcopy)selfrU/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/compilation/counter.pyclone#s zCompilationCounter.clonekwargs)NNNc ksh|}dV|D]%\}}t||t|||ks1J|dt||dt||d|q dS)Nz not as expected, before it is z, after it is z, expected diff is )ritemsgetattr)rroldkvrrrexpect&szCompilationCounter.expectN)rr)__name__ __module__ __qualname__rint__annotations__rrr r r r r rrrrrrrrrrrrrr s             r) r dataclassescollections.abcr contextlibrtypingr dataclassrcompilation_counterrrrrs    &