feat: support aten.any related converters in dynamo

[bowang007]

Description

support aten.any converter

Type of change

• New feature (non-breaking change which adds functionality)

Checklist:

• My code follows the style guidelines of this project (You can use the linters)
• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas and hacks
• I have made corresponding changes to the documentation
• I have added tests to verify my fix or my feature
• New and existing unit tests pass locally with my changes
• I have added the relevant labels to my PR in so that relevant reviewers are notified

[github-actions]

Code conforms to C++ style guidelines

[github-actions]

There are some changes that do not conform to Python style guidelines:

--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py	2024-01-05 04:37:44.368463+00:00
+++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py	2024-01-05 04:39:32.661484+00:00
@@ -2571,10 +2571,11 @@
        src,
        src.dtype,
        force_layer=True,
    )

+
@dynamo_tensorrt_converter(torch.ops.aten.any.default)
@dynamo_tensorrt_converter(torch.ops.aten.any.dim)
@dynamo_tensorrt_converter(torch.ops.aten.any.dims)
def aten_ops_any(
    ctx: ConversionContext,
@@ -2589,6 +2590,6 @@
        SourceIR.ATEN,
        name,
        args[0],
        args_bounds_check(args, 1, replacement=[]),
        args_bounds_check(args, 2, replacement=False),
-    )
\ No newline at end of file
+    )

[narendasan]

Do you have a bool test case? Its probably a common variant of any

[gs-olive]

Added a few suggestions on the implementation. Also needs a rebase to resolve merge conflict.

[github-actions]

There are some changes that do not conform to Python style guidelines:

--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py	2024-01-19 18:47:29.428220+00:00
+++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py	2024-01-19 18:49:20.020601+00:00
@@ -2623,10 +2623,11 @@
        args[0],
        args_bounds_check(args, 1, replacement=None),
        args_bounds_check(args, 2, replacement=False),
    )

+
@dynamo_tensorrt_converter(torch.ops.aten._pdist_forward.default)
@enforce_tensor_types(
    {
        0: (TRTTensor,),
    }

[gs-olive]

Looks good to me; it seems that the test failures are unrelated to the PR.