o
    Ti"                     @   s    d dl mZ G dd deZdS )   )CPUOpBuilderc                       s<   e Zd ZdZdZ fddZdd Zdd Z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/cpu/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   s   ddgS )Nzcsrc/cpu/adam/fused_adam.cppzcsrc/adam/cpu_adam_impl.cppr   r
   r   r   r   sources   s   zFusedAdamBuilder.sourcesc                 C   s   dgS )Nzcsrc/includesr   r
   r   r   r   include_paths   s   zFusedAdamBuilder.include_paths)
__name__
__module____qualname__	BUILD_VARr	   r   r   r   r   __classcell__r   r   r   r   r   	   s    r   N)builderr   r   r   r   r   r   <module>   s   