o
    iw                     @   sp   d Z ddlZddlZddlZddlZddlZdedefddZdd Z	d	d
 Z
edkr6e
ejdd  dS dS )zJ
Generates a wheel metadata.

This tool is NOT intended for end-user use.
    Nlibrary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   a/home/ubuntu/veenaModal/venv/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g d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   )cudnncutensorncclappend)choicesactiondefault   )indent)argparseArgumentParseradd_argumentstr
parse_argsprintr
   dumpsr$   r   targetr   )argsparserparamsr   r   r   main7   s*   


r<   __main__   )__doc__r1   r
   os.pathr   r   r   r4   r   r$   r<   __name__argvr   r   r   r   <module>   s   	