• Wei's avatar
    Enable torchdynamo with torch_tensorrt(fx path) (#17765) · 7ea6ccc2
    Wei authored
    
    
    * enable fx2trt
    
    * Update perf_train_gpu_one.mdx
    
    * Update perf_train_gpu_one.mdx
    
    * add lib check
    
    * update
    
    * format
    
    * update
    
    * fix import check
    
    * fix isort
    
    * improve doc
    
    * refactor ctx manager
    
    * fix isort
    
    * black format
    
    * isort fix
    
    * fix format
    
    * update args
    
    * update black
    
    * cleanups
    
    * Update perf_train_gpu_one.mdx
    
    * code refactor
    
    * code refactor to init
    
    * remove redundancy
    
    * isort
    
    * replace self.args with args
    Co-authored-by: default avatarStas Bekman <stas@stason.org>
    7ea6ccc2
perf_train_gpu_one.mdx 41.6 KB