o
    Tio                     @   sX   d dl Z 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 )    N)__deepspeed__)	OpBuilderc                   @   s$   e Zd Zdd Zdd Zdd ZdS )CPUOpBuilderc                 C   sd   ddl m} dd | |  D }d| |  i}||  | |  || |  |d}|S )Nr   )CppExtensionc                 S   s   g | ]}t j|qS  )ospathabspath).0xr   r   X/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/hpu/builder.py
<listcomp>   s    z(CPUOpBuilder.builder.<locals>.<listcomp>cxx)namesourcesinclude_dirs	librariesextra_compile_args)torch.utils.cpp_extensionr   strip_empty_entriesinclude_pathscxx_argsabsolute_namer   libraries_args)selfExtensionBuilderr   compile_argscpp_extr   r   r   builder   s   zCPUOpBuilder.builderc                 C   s   g d}|S )N)z-O3z-gz-Wno-reorderr   )r   argsr   r   r   r   !   s   zCPUOpBuilder.cxx_argsc                 C   s   g S )Nr   )r   r   r   r   r   %   s   zCPUOpBuilder.libraries_argsN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s    r   )r   
op_builderr   op_builder.builderr   ImportError deepspeed.ops.op_builder.builderr   r   r   r   r   <module>   s   