
    `iw                         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Zd Z	d Z
edk    r e
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                     t           j        dd| d|ddg}t          j        t	          j        |t          j                            t                                        S )Nzinstall_library.py	--library--cudaz--actiondump)cwd)
sys
executablejsonloads
subprocesscheck_outputospathdirname__file__)r   r   commands      x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/tools/_generate_wheel_metadata.py_get_recordsr      sZ    W$FG :RW__X-F-FGGGI I I    c                     | dd}|D ]m}t          ||           D ]F}|d         | k    r8||         }|                    d|          }|||d         |         d         d} nGt          d          |||<   n|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   metadatas	            r   _generate_wheel_metadatar'      s     N  + +"7L99 	D 	DFf~-- /#)::.@'#J#J &(8!'!1-!@!M 
  . BD D D"*wr   c           	         t          j                    }|                    dt          dd           |                    dt          dd           |                    dg dd	g 
           |                    |           }t          t          j        t          |j	        |j
        |j                  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paramss      r   mainr@   7   s    $&&F
sT+  - - -

t@  B B B
 = = =' "  $ $ $ t$$F	$* KMN	
 	
        r   __main__   )__doc__r5   r   os.pathr   r   r
   r8   r   r'   r@   __name__argv r   r   <module>rH      s            



	I# 	IS 	I 	I 	I 	I  0  , zD!"" r   