o
    Ti                     @   s    d dl mZ G dd deZdS )   )SYCLOpBuilderc                       sH   e Zd ZdZdZ fddZdd Zdd Zd	d
 Z fddZ	  Z
S )FusedAdamBuilderDS_BUILD_FUSED_ADAM
fused_adamc                    s   t  j| jd d S )N)name)super__init__NAMEself	__class__ [/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/xpu/fused_adam.pyr      s   zFusedAdamBuilder.__init__c                 C   s   d| j  dS )Nzdeepspeed.ops.adam._op)r	   r
   r   r   r   absolute_name   s   zFusedAdamBuilder.absolute_namec                 C      ddgS )Nz%csrc/xpu/adam/fused_adam_frontend.cppz&csrc/xpu/adam/multi_tensor_adam.dp.cppr   r
   r   r   r   sources      zFusedAdamBuilder.sourcesc                 C   r   )Nzcsrc/xpu/includeszcsrc/xpu/adamr   r
   r   r   r   include_paths   r   zFusedAdamBuilder.include_pathsc                    s   t   }||   S )N)r   cxx_argsversion_dependent_macros)r   argsr   r   r   r      s   
zFusedAdamBuilder.cxx_args)__name__
__module____qualname__	BUILD_VARr	   r   r   r   r   r   __classcell__r   r   r   r   r      s    r   N)builderr   r   r   r   r   r   <module>   s   