o
    Ơi                     @   s  d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ ej	hZ
dZdZdZdZdZdZejZd ZdZdZdZd	ZdZdZdZdZdZd
ZdZdZdZ dZ!ej	ej"ej#ej$ej%ej&hZ'e j()e* ddZ+dZ,dZ-dZ.e j()e* dZ/dZ0dZ1d
Z2dZ3dZ4dZ5d
Z6dZ7dZ8dZ9dZ:dZ;dZ<dZ=dZ>e?e@  ZAe?ejBjCjD  ZEdZFdZGdZHdZIdZJeK dkrd dlLZLeLMe N d  ZOnd dlPZPePQ ZOe j()e* deO dZRdefddZSdS )    N)Device)EngineCapabilitydtypeFi   @i    i         Ttorch_tensorrt_engine_cacheztiming_cache.binl        nonei   Linuxtorch_tensorrt_z/debug_logsreturnc                   C   s   t tj dS )N)gpu_id)r   torchcudacurrent_device r   r   S/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/dynamo/_defaults.pydefault_deviceT   s   r   )Tosplatformtempfiler   torch_tensorrt._Devicer   torch_tensorrt._enumsr   r   f32ENABLED_PRECISIONSDEVICEDISABLE_TF32ASSUME_DYNAMIC_SHAPE_SUPPORTDLA_LOCAL_DRAM_SIZEDLA_GLOBAL_DRAM_SIZEDLA_SRAM_SIZESTANDARDENGINE_CAPABILITYWORKSPACE_SIZEMIN_BLOCK_SIZEPASS_THROUGH_BUILD_FAILURESMAX_AUX_STREAMSNUM_AVG_TIMING_ITERSVERSION_COMPATIBLEOPTIMIZATION_LEVELSPARSE_WEIGHTSTRUNCATE_DOUBLEUSE_PYTHON_RUNTIMEUSE_FAST_PARTITIONER"ENABLE_EXPERIMENTAL_DECOMPOSITIONSREQUIRE_FULL_COMPILATIONDRYRUNHARDWARE_COMPATIBLEf16bf16i8f8f4SUPPORTED_KERNEL_PRECISIONSpathjoin
gettempdirTIMING_CACHE_PATHLAZY_ENGINE_INITCACHE_BUILT_ENGINESREUSE_CACHED_ENGINESENGINE_CACHE_DIRENGINE_CACHE_SIZECUSTOM_ENGINE_CACHEUSE_EXPLICIT_TYPINGUSE_FP32_ACCREFIT_IDENTICAL_ENGINE_WEIGHTSSTRIP_ENGINE_WEIGHTSIMMUTABLE_WEIGHTSENABLE_WEIGHT_STREAMING ENABLE_CROSS_COMPILE_FOR_WINDOWSTILING_OPTIMIZATION_LEVELL2_LIMIT_FOR_TILINGUSE_DISTRIBUTED_MODE_TRACEOFFLOAD_MODULE_TO_CPUENABLE_AUTOCASTAUTOCAST_LOW_PRECISION_TYPEsetstrAUTOCAST_EXCLUDED_NODESfxnodeTargetAUTOCAST_EXCLUDED_OPSAUTOCAST_MAX_OUTPUT_THRESHOLDAUTOCAST_MAX_DEPTH_OF_REDUCTIONAUTOCAST_CALIBRATION_DATALOADERENABLE_RESOURCE_PARTITIONINGCPU_MEMORY_BUDGETsystempwdgetpwuidgetuidcurrent_usergetpassgetuserDEBUG_LOGGING_DIRr   r   r   r   r   <module>   s    

