I'm not sure if this is intended, but it seemed odd to me that there is logic inside both fast and global partitioners (see snippets 1 and 2) with require_full_compilation but the compile_module never feeds the passes the flag as parameter (see snippet 3).
I'm not sure if this is intended, but it seemed odd to me that there is logic inside both fast and global partitioners (see snippets 1 and 2) with
require_full_compilation
but thecompile_module
never feeds the passes the flag as parameter (see snippet 3).Snippets
1. https://github.com/pytorch/TensorRT/blob/fa02fd3e3a85a6042d11a00cd386f6b69c1d6c4b/py/torch_tensorrt/dynamo/partitioning/_adjacency_partitioner.py#L206-L209
2. https://github.com/pytorch/TensorRT/blob/fa02fd3e3a85a6042d11a00cd386f6b69c1d6c4b/py/torch_tensorrt/dynamo/partitioning/_global_partitioner.py#L68-L71
3. https://github.com/pytorch/TensorRT/blob/fa02fd3e3a85a6042d11a00cd386f6b69c1d6c4b/py/torch_tensorrt/dynamo/_compiler.py#L371-L388