a
    d                     @   s   d dl Z d dlmZ d dlmZmZ d dlZd dlmZ ddl	m
Z
 ddlmZ edd	 Zed
ddd Ze
edZeded e
eedd e jedddZeded e
edZeded dS )    N)import_module)#min_cut_rematerialization_partitionnop)
ts_compile   )aot_autograd)register_debug_backendc                 C   s   | S N ZgmZfake_tensor_inputsr
   r
   i/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_dynamo/backends/debugging.pyeager   s    r   ts)namec                 C   s   t j| S r	   )torchZjitscriptr   r
   r
   r   torchscript   s    r   )fw_compiler	aot_eager)r   Zcompiler_fnc                   C   s   t d S )Nztorch._inductor.compile_fx)r   Zselect_decomp_tabler
   r
   r
   r   <lambda>)   s   r   Zinductor)compiler)r   Zbw_compilerZdecompositionsZpartition_fnaot_eager_decomp_partitionaot_ts)	functools	importlibr   Zfunctorch.compiler   r   r   Ztorch._functorch.compilersr   commonr   registryr   Zregister_backendr   r   r   partialr   r   r
   r
   r
   r   <module>   s2   



