Moving elementwise core to impl - rsqrt (FX Converter Refactor [9/N]) <Target: converter_reorg_elementwise> #1905

apbose · 2023-05-10T05:57:08Z

github-actions

Code conforms to C++ style guidelines

github-actions

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

--- py/torch_tensorrt/fx/converters/aten_ops_converters.py	2023-05-10 05:57:24.442200 +0000
+++ py/torch_tensorrt/fx/converters/aten_ops_converters.py	2023-05-10 05:57:43.318756 +0000
@@ -325,15 +325,15 @@
    target: Target,
    args: Tuple[Argument, ...],
    kwargs: Dict[str, Argument],
    name: str,
) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    
+
    return rsqrt(
-        network, 
-        target, 
-        SourceIR.ATEN, 
+        network,
+        target,
+        SourceIR.ATEN,
        name,
        args[0],
    )


--- py/torch_tensorrt/fx/test/converters/aten_op/test_rsqrt_aten.py	2023-05-10 05:57:24.450200 +0000
+++ py/torch_tensorrt/fx/test/converters/aten_op/test_rsqrt_aten.py	2023-05-10 05:57:45.176604 +0000
@@ -24,6 +24,6 @@
            expected_ops={torch.ops.aten.rsqrt.default},
        )


if __name__ == "__main__":
-    run_tests()
\ No newline at end of file
+    run_tests()

github-actions

Code conforms to Python style guidelines

github-actions

Code conforms to C++ style guidelines

github-actions

Code conforms to C++ style guidelines

github-actions

Code conforms to Python style guidelines

github-actions

Code conforms to Python style guidelines

github-actions

Code conforms to C++ style guidelines

github-actions

Code conforms to C++ style guidelines

github-actions

Code conforms to Python style guidelines

github-actions

Code conforms to C++ style guidelines

github-actions

Code conforms to Python style guidelines

Revert "fix: Add automatic type promotion for FX ops" This reverts commit f1f3716. Revert "Moved clamp to impl" This reverts commit df401dd. Revert "aten::unsqueeze impl refactor" This reverts commit b424735. Revert "Converter reorg and where operator" This reverts commit b4da15e. Revert "converter reorg and matmul" This reverts commit 7551eee. Revert "converter reorg and slice" This reverts commit 9bbdc9e. Revert "Converter reorg and select operation" This reverts commit fb70253. Revert "Converter reorg and squeeze operator" This reverts commit 294545c. Revert "Converter reorg and gelu" This reverts commit 37d1168. Revert "Converter reorg and softmax operation" This reverts commit 1ba6d13. Revert "layer_norm converter" This reverts commit e0b34b1. Revert "Converter reorg batch norm" This reverts commit 59354e5. Revert "Converter reorg and rsub" This reverts commit db15d27. Revert "Converter reorg fmod" This reverts commit ce3fa67. Revert "Moving elementwise core to impl - rsqrt (FX Converter Refactor [9/N]) <Target: converter_reorg_elementwise> (#1905)" This reverts commit 7158ca5. Revert "refactor: Moving elementwise and unary core to impl" This reverts commit 45e43ca.

refactor: Moving elementwise and unary core to impl Signed-off-by: Naren Dasan <naren@narendasan.com> new file: ../converters/impl/unary/base.py Moving elementwise core to impl - rsqrt (FX Converter Refactor [9/N]) <Target: converter_reorg_elementwise> (#1905) Converter reorg fmod Converter reorg and rsub Rsub error fixes and linting error fixed Rsub test case to include different inputs Converter reorg batch norm batch norm error fix and linting issue error fix layer_norm converter Layer norm linting correction ops file correction fixing lint Acc_ops layer_norm correction Converter reorg and softmax operation softmax linting error fix Converter reorg and gelu Linting error Converter reorg and squeeze operator Correcting squeeze operator implementation, linting error and acc squeeze test Adding the condition to convert dim to int and removing the comment Converter reorg and select operation select operation correction and linting changes converter reorg and slice converter reorg slice op Correcting linting error and slice changes Correcting the slice operation converter reorg and matmul Matmul issue fixes and lint error check moving matmul to individual file Converter reorg and where operator adding where aten op aten::where correction and linting error changes aten::unsqueeze impl refactor Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> Moved clamp to impl Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> fixed method name Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> fix: Add automatic type promotion for FX ops - Implement functionality to cast tensors to alternative types - Add functionality to elementwise ops to promote types and perform necessary casts - Address issues in FX ops where mixed-precision computations can cause errors - Add test cases to validate fix