o
    TiR                     @   s    d dl mZ G dd deZdS )   )NPUOpBuilderc                       s@   e Zd ZdZdZ fddZdd Zdd Z f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/npu/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   }|dg7 }|S )Nzcsrc/includes)r   include_paths)r   argsr   r   r   r      s   

zCPUAdamBuilder.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   