oops, forgot to specify expected results for lit test

daveliddell · Feb 6, 2024 · b46c489 · b46c489
1 parent ea3b1b4
commit b46c489
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/Dialect/Torch/canonicalize.mlir b/test/Dialect/Torch/canonicalize.mlir
@@ -2196,15 +2196,16 @@ func.func @torch.aten.detach$canonicalize(%arg0: !torch.tensor<[1],f32>) -> !tor
 
 // CHECK-LABEL:   func.func @torch.aten.index_select$noop(
 // CHECK-SAME:      %[[ARG:.*]]: !torch.vtensor<[1,2,3],si64>
-// CHECK-NEXT:  foo
+// CHECK-NEXT:      return %[[ARG]] : !torch.vtensor<[1,2,3],si64>
 func.func @torch.aten.index_select$noop(%arg0 : !torch.vtensor<[1,2,3],si64>, %arg1 : !torch.int, %arg2 : !torch.vtensor<[1],si64>) -> (!torch.vtensor<[1,2,3],si64>) {
   %0 = torch.aten.index_select %arg0, %arg1, %arg2 : !torch.vtensor<[1,2,3],si64>, !torch.int, !torch.vtensor<[1],si64> -> !torch.vtensor<[1,2,3],si64>
   return %0 : !torch.vtensor<[1,2,3],si64>
 }
 
 // CHECK-LABEL:   func.func @torch.aten.index_select$const(
 // CHECK-SAME:      %[[ARG:.*]]: !torch.vtensor<[1,2,3],si64>
-// CHECK-NEXT:  foo
+// CHECK-NEXT:      %[[RES:.*]] = torch.vtensor.literal(dense<60> : tensor<1xsi64>) : !torch.vtensor<[1],si64>
+// CHECK-NEXT:      return %[[RES]] : !torch.vtensor<[1],si64>
 func.func @torch.aten.index_select$const(%arg0 : !torch.vtensor<[1,2,3],si64>, %arg1 : !torch.int, %arg2 : !torch.vtensor<[1],si64>) -> (!torch.vtensor<[1],si64>) {
   %tensor = torch.vtensor.literal(dense<[10,20,30,40,50,60,70,80,90,100]> : tensor<10xsi64>) : !torch.vtensor<[10],si64>
   %dim = torch.constant.int 0