From 7669ee4e4af23ff96a47a8f88b216f2318df7f0f Mon Sep 17 00:00:00 2001 From: Sean Silva Date: Thu, 28 Apr 2022 13:36:42 +0000 Subject: [PATCH] Revert "[MLIR][TORCH] Add E2E support for aten.ne.float_int op" This reverts commit 51dd4625927329f528b549faec90ff0be2d93392. --- .../Dialect/Torch/IR/GeneratedTorchOps.td | 24 ------------------- lib/Conversion/TorchToStd/TorchToStd.cpp | 5 +--- .../jit_ir/build_tools/torch_ods_gen.py | 1 - .../test_suite/scalar_comparison.py | 20 ---------------- test/Conversion/TorchToStd/basic.mlir | 14 ----------- 5 files changed, 1 insertion(+), 63 deletions(-) diff --git a/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td b/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td index 844ddca29..b9d821ef4 100644 --- a/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td +++ b/include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td @@ -6667,30 +6667,6 @@ def Torch_AtenGeFloatIntOp : Torch_Op<"aten.ge.float_int", [ }]; } -def Torch_AtenNeFloatIntOp : Torch_Op<"aten.ne.float_int", [ - AllowsTypeRefinement, - HasValueSemantics, - ReadOnly - ]> { - let summary = "Generated op for `aten::ne.float_int : (float, int) -> (bool)`"; - let arguments = (ins - Torch_FloatType:$a, - Torch_IntType:$b - ); - let results = (outs - Torch_BoolType:$result - ); - let hasCustomAssemblyFormat = 1; - let extraClassDefinition = [{ - ParseResult AtenNeFloatIntOp::parse(OpAsmParser &parser, OperationState &result) { - return parseDefaultTorchOp(parser, result, 2, 1); - } - void AtenNeFloatIntOp::print(OpAsmPrinter &printer) { - printDefaultTorchOp(printer, *this, 2, 1); - } - }]; -} - def Torch_Aten__And__BoolOp : Torch_Op<"aten.__and__.bool", [ AllowsTypeRefinement, HasValueSemantics, diff --git a/lib/Conversion/TorchToStd/TorchToStd.cpp b/lib/Conversion/TorchToStd/TorchToStd.cpp index 718c44c7a..e207c99c2 100644 --- a/lib/Conversion/TorchToStd/TorchToStd.cpp +++ b/lib/Conversion/TorchToStd/TorchToStd.cpp @@ -211,16 +211,13 @@ public: patterns.add< ConvertAtenIntComparisonOp>( typeConverter, context); - target.addIllegalOp(); + target.addIllegalOp(); patterns.add< ConvertAtenFloatComparisonOp>( typeConverter, context); patterns.add>( typeConverter, context); - patterns.add>( - typeConverter, context); target.addIllegalOp(); patterns.add(typeConverter, context); diff --git a/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/torch_ods_gen.py b/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/torch_ods_gen.py index e0d498d85..65859bf14 100644 --- a/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/torch_ods_gen.py +++ b/python/torch_mlir/dialects/torch/importer/jit_ir/build_tools/torch_ods_gen.py @@ -498,7 +498,6 @@ def emit_ops(emitter_td: TextEmitter, registry: Registry): emit("aten::lt.float : (float, float) -> (bool)", has_folder=True) emit("aten::lt.float_int : (float, int) -> (bool)") emit("aten::ge.float_int : (float, int) -> (bool)") - emit("aten::ne.float_int : (float, int) -> (bool)") emit("aten::__and__.bool : (bool, bool) -> (bool)") emit("aten::ne.bool : (bool, bool) -> (bool)", has_folder=True) emit("aten::__is__ : (t1, t2) -> (bool)", has_folder=True) diff --git a/python/torch_mlir_e2e_test/test_suite/scalar_comparison.py b/python/torch_mlir_e2e_test/test_suite/scalar_comparison.py index dc9d57564..ff60e3821 100644 --- a/python/torch_mlir_e2e_test/test_suite/scalar_comparison.py +++ b/python/torch_mlir_e2e_test/test_suite/scalar_comparison.py @@ -108,23 +108,3 @@ class GeFloatIntModule(torch.nn.Module): @register_test_case(module_factory=lambda: GeFloatIntModule()) def GeFloatIntModule_basic(module, tu: TestUtils): module.forward(torch.randn(()), torch.randint(-100, 100, ())) - -# ============================================================================== - -class NeFloatIntModule(torch.nn.Module): - def __init__(self): - super().__init__() - - @export - @annotate_args([ - None, - ([], torch.float64, True), - ([], torch.int64, True), - ]) - def forward(self, lhs, rhs): - return float(lhs) != int(rhs) - - -@register_test_case(module_factory=lambda: NeFloatIntModule()) -def NeFloatIntModule_basic(module, tu: TestUtils): - module.forward(torch.randn(()), torch.randint(-100, 100, ())) diff --git a/test/Conversion/TorchToStd/basic.mlir b/test/Conversion/TorchToStd/basic.mlir index 5b3ca1792..3c9d16db2 100644 --- a/test/Conversion/TorchToStd/basic.mlir +++ b/test/Conversion/TorchToStd/basic.mlir @@ -193,17 +193,3 @@ func @torch.aten.ge.float_int(%arg0: !torch.float, %arg1: !torch.int) -> !torch. %0 = torch.aten.ge.float_int %arg0, %arg1 : !torch.float, !torch.int -> !torch.bool return %0 : !torch.bool } - -// CHECK-LABEL: func @torch.aten.ne.float_int( -// CHECK-SAME: %[[LHS:.*]]: !torch.float, -// CHECK-SAME: %[[RHS:.*]]: !torch.int) -> !torch.bool { -// CHECK: %[[LHS_F64:.*]] = torch_c.to_f64 %[[LHS]] -// CHECK: %[[RHS_I64:.*]] = torch_c.to_i64 %[[RHS]] -// CHECK: %[[RHS_F64:.*]] = arith.sitofp %[[RHS_I64]] : i64 to f64 -// CHECK: %[[CMP:.*]] = arith.cmpf une, %[[LHS_F64]], %[[RHS_F64]] : f64 -// CHECK: %[[CMP_TORCH_BOOL:.*]] = torch_c.from_i1 %[[CMP]] -// CHECK: return %[[CMP_TORCH_BOOL]] : !torch.bool -func @torch.aten.ne.float_int(%arg0: !torch.float, %arg1: !torch.int) -> !torch.bool { - %0 = torch.aten.ne.float_int %arg0, %arg1 : !torch.float, !torch.int -> !torch.bool - return %0 : !torch.bool -}