o
    X۷i                     @  st   d Z ddlmZ ddlZddlZddlZddlZddlZdddZ	d	d
 Z
dd Zedkr8eejdd  dS dS )zJ
Generates a wheel metadata.

This tool is NOT intended for end-user use.
    )annotationsNlibrarystrcudac                 C  s2   t jdd| d|ddg}ttj|tjt	dS )Nzinstall_library.py	--library--cudaz--actiondump)cwd)
sys
executablejsonloads
subprocesscheck_outputospathdirname__file__)r   r   command r   Z/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/tools/_generate_wheel_metadata.py_get_records   s   r   c           	      C  st   | dd}|D ]0}t || D ] }|d | kr.|| }|d|}|||d | d d} nqtd|||< q|S )	Npip)r   	packagingr   min_pypi_versionassets	filenames)versionr   r   z0Specified library/CUDA combination not supported)r   getRuntimeError)	cuda_versiontarget_system	librarieswheel_metadatar   recordr   r   metadatar   r   r   _generate_wheel_metadata!   s&   

r&   c                 C  sp   t  }|jdtddd |jdtddd |jddd	gd
g d || }ttjt|j	|j
|jdd d S )Nr   TzCUDA version)typerequiredhelpz--targetz!Target system (e.g., Linux:x86_64r   cutensorncclappend)choicesactiondefault   )indent)argparseArgumentParseradd_argumentr   
parse_argsprintr   dumpsr&   r   targetr   )argsparserparamsr   r   r   main9   s*   


r<   __main__   )r   r   r   r   )__doc__
__future__r   r2   r   os.pathr   r   r
   r   r&   r<   __name__argvr   r   r   r   <module>   s   
