o
    Tio                     @   s    d dl mZ G dd deZdS )   )CUDAOpBuilderc                       sF   e Zd ZdZdZd fdd	Zdd Zdd	 Zd
d Zdd Z	  Z
S )RandomLTDBuilderDS_BUILD_RANDOM_LTD
random_ltdNc                    s$   |d u r| j n|}t j|d d S )N)name)NAMEsuper__init__)selfr   	__class__ W/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/random_ltd.pyr	      s   zRandomLTDBuilder.__init__c                 C   s   d| j  dS )Nzdeepspeed.ops._op)r   r
   r   r   r   absolute_name   s   zRandomLTDBuilder.absolute_namec                 C   s   |   sdgS g S )Nz-lcurand)is_rocm_pytorchr   r   r   r   extra_ldflags   s   zRandomLTDBuilder.extra_ldflagsc                 C   s   g dS )N)zcsrc/random_ltd/pt_binding.cppz!csrc/random_ltd/gather_scatter.cuz#csrc/random_ltd/slice_attn_masks.cuzcsrc/random_ltd/token_sort.cur   r   r   r   r   sources   s   zRandomLTDBuilder.sourcesc                 C   s
   dg}|S )Nzcsrc/includesr   )r
   includesr   r   r   include_paths    s   zRandomLTDBuilder.include_paths)N)__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   