o
    Ti
                     @   sP   zd dl mZ d dlmZ W n ey   d dlmZ Y nw G dd deZdS )    )__deepspeed__)	OpBuilderc                   @   s$   e Zd Zdd Zdd Zdd ZdS )SDAAOpBuilderc                 C   sV   ddl m} d| |  i}||  | |  | |  | |  |d}|S )Nr   )CppExtensioncxx)namesourcesinclude_dirs	librariesextra_compile_args)torch.utils.cpp_extensionr   strip_empty_entriescxx_argsabsolute_namer   include_pathslibraries_args)selfExtensionBuildercompile_argscpp_ext r   Y/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/sdaa/builder.pybuilder+   s   zSDAAOpBuilder.builderc                 C   s   g dS )N)z-O3z-gz-Wno-reorderr   r   r   r   r   r   8   s   zSDAAOpBuilder.cxx_argsc                 C   s   g S )Nr   r   r   r   r   r   ;   s   zSDAAOpBuilder.libraries_argsN)__name__
__module____qualname__r   r   r   r   r   r   r   r   )   s    r   N)
op_builderr   op_builder.builderr   ImportError deepspeed.ops.op_builder.builderr   r   r   r   r   <module>   s   