o
    il                     @   s`   d dl mZ d dlmZ d dlmZmZmZmZm	Z	 d dl
mZ defddZdefdd	Zd
S )    DynamoOptimizationResult)innermost_fn)ListCallableDictUnionSet)CodeTypeoriginal_codec                 C   s   t | d |}|S )Nr   )r   moduleresult r   S/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/depyf/explain/__init__.py_extract_artifacts	   s   r   c                 C   s.   ddl m} |d sJ dt| |}| S )Nr   )datais_inside_prepare_debugz5`dump_src` must be used inside `depyf.prepare_debug`.)depyf.explain.global_variablesr   r   to_src)r   r   r   	artifactsr   r   r   dump_src   s   
r   N)depyf.explain.utilsr   torch._dynamo.eval_framer   typingr   r   r   r   r	   typesr
   r   r   r   r   r   r   <module>   s    