Zewen (Evan) Li
Zewen (Evan) Li
Hi, Robinson! Thanks for your amazing work and dataset! I found some ``csv`` files in ``sphereface_rfiw_baseline/val/pairs_withlabel-CSV/``, but I cannot relate them to FIW dataset. So can you tell me how...
# Description Support `adaptive_avg_pool1d` dynamo_converter. Fixes #2603 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My code follows the style guidelines of...
# Description Support `adaptive_avg_pool2d` and `adaptive_avg_pool3d`. Fixes #2611 #2612 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My code follows the style...
Current `index` converter only supports `int` type while cannot handle `bool` mask indices. We need to expand index support to boolean masks, since this is a valid usage in Torch...
# Description Support 1d ITensor offsets for embedding_bag converter. Note that this is only for 1d inputs. There's a bug that same offsets with different types (tensor or ITensor) when...
## Bug Description CUDNN is not to be the dependency in the main branch. ## Environment - Torch-TensorRT Version: main branch - PyTorch Version: 2.3 - TensorRT Version: 10.0.1.6
## Bug Description The failures were shown in CI (https://github.com/pytorch/TensorRT/actions/runs/8863273616/job/24391143167?pr=2781): ``` =========================== short test summary info ============================ FAILED models/test_dyn_models.py::test_base_dynamic - torch.fx.experimental.symbolic_shapes.ConstraintViolationError: Constraints violated (L['x'].size()[0])! For more information, run with TORCH_LOGS="+dynamic"....
# Description Added a pre-AOT lowering pass to remove `detach` ops Fixes #2657 ## Type of change - New feature (non-breaking change which adds functionality) # Checklist: - [x] My...