o
    Ti                     @   s   d dl Z d dlZd dlZd dlZddlmZmZ dZzd dlZ	dZ
W n ey-   dZ
Y nw g Ze je Zdd ZeejejgD ]2\ZZZed	krved
krvejde e
dZe D ]Zedruedkruedkrueeeje< q_qDdS )    N   ) get_default_compute_capabilities	OpBuilderTzdeepspeed.ops.op_builder
op_builderc                    s4   t dkr fdd}|S ddlm} |  }|S )Nr   c                     s   ddl m}  |   }|S )Nr   get_accelerator)deepspeed.acceleratorr   create_op_builder)r   buildermember_name U/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/__init__.py_builder    s   z!builder_closure.<locals>._builderr   r   )op_builder_dirr   r   get_op_builder)r   r   r   r
   r   r   r   builder_closure   s   r   all_opsr
   .)packageBuilderr   CUDAOpBuilder)sysospkgutil	importlibr
   r   r   __deepspeed__deepspeed.ops.op_builder	deepspeedr   ImportError__op_builders__modules__name__this_moduler   iter_modulespathdirname__file___module_nameimport_modulemodule__dir__r   endswith__dict__r   r   r   r   <module>   s.   
 