o i @sddlZddlZddlZddlZdZzddlZWn eefy#YnwdZdZd dkr0dZesej dre e ejd dZd d Zd ed dded dgd ed dded dgdgdgddZeD]ZeeZesuqlejes}Jeeql[[ddlTdZddZddZee_dde_ee_ee_ee _ee _ee!_ee!_ee"_ee"_ddksddkree#_ee#_ee$_ee$_ee%_ee%_ee&_ee&_ej'e(_'ej'j)*D] \Z+Z,e-e(e+e,qddZ.d d!Z/d"d#Z0e1d$de2_3dS)%NFTnvinfer tensorrt_rtx_14winTRT_LOG_FOUND_DLLScCstjdtjj}|tjtjttj d|D]}tj||}tj |r;t r7t d|d||Sq | drDdStd||) NPATHz tensorrt.libszFound z in path: Fnvinfer_pluginnvinfer_builder_resourcezGCould not find: {:}. Is it on your PATH? Note: Paths searched were: {:})osenvironsplitpathpathsepappendjoindirname__file__pardirisfilelog_found_dllsprint startswithFileNotFoundErrorformat)namepathsrlibpathrN/home/ubuntu/.local/lib/python3.10/site-packages/tensorrt_bindings/__init__.pyfind_lib(s    r nvinfer_10z.dllznvinfer_plugin_10.dllnvonnxparser_10znvinfer_builder_resource_10.dllznvinfer_dispatch_10.dllznvinfer_lean_10.dll)tensorrtrtensorrt_dispatch tensorrt_leanr#)*z10.14.1.48.post1cCstdt|S)NzzContext managers for TensorRT types are deprecated. Memory will be freed automatically when the reference count reaches 0.)warningswarnDeprecationWarningthisrrr common_enterfs r-cCsdS)z Context managers are deprecated and have no effect. Objects are automatically freed when the reference count reaches 0. Nrr,exc_type exc_value tracebackrrr common_exitosr2cCsdSNrr.rrrysr4cCsd}|D]}||9}q|S)a Computes the volume of an iterable. :arg iterable: Any python iterable, including a :class:`Dims` object. :returns: The volume of the iterable. This will return 1 for empty iterables, as a scalar has an empty shape and the volume of a tensor with empty shape is 1. r&r)iterablevolelemrrrvolumes r8cCsNddl}t|jt|jt|jt|jt|jt|jt|ji}||vr#||St d)z Returns the numpy-equivalent of a TensorRT :class:`DataType` . :arg trt_type: The TensorRT data type to convert. :returns: The equivalent numpy type. rNzDCould not resolve TensorRT datatype to an equivalent numpy datatype.) numpyfloat32float16int8int32int64boolbool_uint8 TypeError)trt_typenpmappingrrrnptypes rFcCsDtdtdtdtdtdtdtdtdtdt dt di }||vr ||SdS)z Returns the size in bytes of this :class:`DataType`. The returned size is a rational number, possibly a `Real` denoting a fraction of a byte. :arg trt_type: The TensorRT data type. :returns: The size of the type. r&g?N) r:r;bfloat16r<r=r>r?rAfp8int4fp4)rCrErrr _itemsizes  rNcCst|Sr3)rNr+rrrr4s)4ctypesr sysr(_libs_wheel_imported tensorrt_libs ImportErrorModuleNotFoundError_trt_lib_suffixstripplatformrr?intr getrr LIBRARIESliblib_pathrrCDLLr# __version__r-r2ILogger __enter____exit__ ICudaEngineIExecutionContextRuntime IHostMemoryBuilderINetworkDefinition OnnxParserIBuilderConfigSeverityLogger __members__itemsattrvaluesetattrr8rFrNpropertyDataTypeitemsizerrrrs