Skip to content

Commit 47f7f84

Browse files
committed
update lit tests and xfails
1 parent 6310c63 commit 47f7f84

File tree

2 files changed

+4
-11
lines changed

2 files changed

+4
-11
lines changed

projects/pt1/e2e_testing/xfail_sets.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2605,9 +2605,6 @@
26052605
"BernoulliTensorModule_basic",
26062606
# Failure - onnx_lowering: onnx.ReduceProd
26072607
"ReduceProdDimIntFloatModule_basic",
2608-
# Failure - onnx_lowering: onnx.Resize
2609-
"UpSampleNearest2dDynamicSize_basic",
2610-
"UpSampleNearest2dStaticSize_basic",
26112608
# Failure - onnx_lowering: onnx.ScatterElements
26122609
"ScatterReduceFloatMaxModuleIncludeSelf",
26132610
"ScatterReduceFloatMinModuleIncludeSelf",

test/Conversion/TorchToLinalg/resize.mlir

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,13 @@
44
func.func @test_resize_sizes_linear(%arg0: !torch.vtensor<[1,1,2,4],f32>, %arg1: !torch.vtensor<[4]
55
,si64>) -> !torch.vtensor<[?,?,?,?],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 19 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} {
66
// CHECK: %[[generic:.*]] = linalg.generic
7-
// CHECK: %[[c2_i64:.*]] = arith.constant 2 : i64
8-
// CHECK: %[[c4_i64:.*]] = arith.constant 4 : i64
97
// CHECK: %[[cst:.*]] = arith.constant 1.001000e+00 : f32
108
// CHECK: %[[cst_4:.*]] = arith.constant 1.000000e+00 : f32
119
// CHECK: %[[cst_5:.*]] = arith.constant 5.000000e-01 : f32
1210
// CHECK: %[[cst_6:.*]] = arith.constant 0.000000e+00 : f32
1311
// CHECK: %[[x13:.*]] = linalg.index 2 : index
1412
// CHECK: %[[x14:.*]] = linalg.index 3 : index
15-
// CHECK: %[[x15:.*]] = arith.sitofp %[[c2_i64]] : i64 to f32
13+
// CHECK: %[[x15:.*]] = arith.sitofp %[[c2_i64:.*]] : i64 to f32
1614
// CHECK: %[[x16:.*]] = arith.sitofp %[[x6:.*]] : i64 to f32
1715
// CHECK: %[[x17:.*]] = arith.divf %[[x16]], %[[x15]] : f32
1816
// CHECK: %[[x18:.*]] = arith.index_cast %[[x13]] : index to i64
@@ -23,7 +21,7 @@ func.func @test_resize_sizes_linear(%arg0: !torch.vtensor<[1,1,2,4],f32>, %arg1:
2321
// CHECK: %[[x23:.*]] = arith.maximumf %[[x22]], %[[cst_6]] : f32
2422
// CHECK: %[[x24:.*]] = arith.subf %[[x15]], %[[cst]] : f32
2523
// CHECK: %[[x25:.*]] = arith.minimumf %[[x23]], %[[x24]] : f32
26-
// CHECK: %[[x26:.*]] = arith.sitofp %[[c4_i64]] : i64 to f32
24+
// CHECK: %[[x26:.*]] = arith.sitofp %[[c4_i64:.*]] : i64 to f32
2725
// CHECK: %[[x27:.*]] = arith.sitofp %[[x7:.*]] : i64 to f32
2826
// CHECK: %[[x28:.*]] = arith.divf %[[x27]], %[[x26]] : f32
2927
// CHECK: %[[x29:.*]] = arith.index_cast %[[x14]] : index to i64
@@ -96,12 +94,10 @@ func.func @test_resize_sizes_linear(%arg0: !torch.vtensor<[1,1,2,4],f32>, %arg1:
9694

9795
func.func @test_resize_sizes_nearest(%arg0: !torch.vtensor<[1,1,2,4],f32>, %arg1: !torch.vtensor<[4],si64>) -> !torch.vtensor<[?,?,?,?],f32> attributes {torch.onnx_meta.ir_version = 7 : si64, torch.onnx_meta.opset_version = 19 : si64, torch.onnx_meta.producer_name = "backend-test", torch.onnx_meta.producer_version = ""} {
9896
// CHECK: %[[GENERIC:.*]] = linalg.generic
99-
// CHECK: %[[c2_i64:.*]] = arith.constant 2 : i64
100-
// CHECK: %[[c4_i64:.*]] = arith.constant 4 : i64
10197
// CHECK: %[[x13:.*]] = linalg.index 2 : index
10298
// CHECK: %[[x14:.*]] = linalg.index 3 : index
103-
// CHECK: %[[x15:.*]] = arith.sitofp %[[c2_i64]] : i64 to f32
104-
// CHECK: %[[x16:.*]] = arith.sitofp %[[c4_i64]] : i64 to f32
99+
// CHECK: %[[x15:.*]] = arith.sitofp %[[c2_i64:.*]] : i64 to f32
100+
// CHECK: %[[x16:.*]] = arith.sitofp %[[c4_i64:.*]] : i64 to f32
105101
// CHECK: %[[x19:.*]] = arith.sitofp %[[x6:.*]] : i64 to f32
106102
// CHECK: %[[x20:.*]] = arith.sitofp %[[x7:.*]] : i64 to f32
107103
// CHECK: %[[x21:.*]] = arith.divf %[[x19]], %[[x15]] : f32

0 commit comments

Comments
 (0)