o
    TÃiÇ  ã                   @   sì   d dl Z d dlZd dlZzd dlmZ W n ey#   d dlmZ Y nw eƒ  ¡ Ze e¡Z	g Z
e e j e	j¡g¡D ]-\ZZZedkrhedkrhe d ee¡¡Ze ¡ D ]Ze d¡rgeƒ  e¡Ze
 e¡ qUq;dd„ e
D ƒZeƒ jZdS )	é    N)Úget_acceleratorÚall_opsÚbuilderz{}.{}ÚBuilderc                 C   s   i | ]
}|d ur|j |“qS )N)Úname)Ú.0Úop© r	   úT/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/all_ops.pyÚ
<dictcomp>    s    r   )ÚosÚpkgutilÚ	importlibÚacceleratorr   ÚImportErrorÚdeepspeed.acceleratorÚop_builder_dirÚimport_moduleÚop_builder_moduleÚ__op_builders__Úiter_modulesÚpathÚdirnameÚ__file__Ú_Úmodule_nameÚformatÚmoduleÚ__dir__Úmember_nameÚendswithÚcreate_op_builderr   ÚappendÚALL_OPSÚ_nameÚaccelerator_namer	   r	   r	   r
   Ú<module>   s*   ÿ

 

€