o
    Tiq                     @   s    d dl mZ G dd deZdS )   )CPUOpBuilderc                       sH   e Zd ZdZdZ fddZdd Zdd Z fd	d
Zdd Z	  Z
S )CPUAdamBuilderDS_BUILD_CPU_ADAMcpu_adamc                    s   t  j| jd d S )N)name)super__init__NAMEself	__class__ Y/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/cpu/cpu_adam.pyr      s   zCPUAdamBuilder.__init__c                 C   s   d| j  dS )Nzdeepspeed.ops.adam._op)r	   r
   r   r   r   absolute_name   s   zCPUAdamBuilder.absolute_namec                 C   s   ddgS )Nzcsrc/adam/cpu_adam.cppzcsrc/adam/cpu_adam_impl.cppr   r
   r   r   r   sources   s   zCPUAdamBuilder.sourcesc                    s   t   }|S )N)r   libraries_args)r   argsr   r   r   r      s   
zCPUAdamBuilder.libraries_argsc                 C   s   dgS )Nzcsrc/includesr   r
   r   r   r   include_paths   s   zCPUAdamBuilder.include_paths)__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   