From e0af1523bccd8a75c2adee12bac8103fe8da36bb Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Mon, 15 Nov 2021 12:04:45 +0100 Subject: [PATCH] cleanup(compiler/test): Formatting of mlir test string --- .../unittest/end_to_end_jit_encrypted_tensor.cc | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/compiler/tests/unittest/end_to_end_jit_encrypted_tensor.cc b/compiler/tests/unittest/end_to_end_jit_encrypted_tensor.cc index 4cf2325a2..72004ed6b 100644 --- a/compiler/tests/unittest/end_to_end_jit_encrypted_tensor.cc +++ b/compiler/tests/unittest/end_to_end_jit_encrypted_tensor.cc @@ -65,9 +65,8 @@ func @main(%t: tensor<2x10x!HLFHE.eint<6>>, %i: index, %j: index) -> TEST(End2EndJit_EncryptedTensor_2D, extract_slice) { mlir::zamalang::JitCompilerEngine::Lambda lambda = checkedJit(R"XXX( func @main(%t: tensor<2x10x!HLFHE.eint<6>>) -> tensor<1x5x!HLFHE.eint<6>> { - %r = tensor.extract_slice %t[1, 5][1, 5][1, 1] : - tensor<2x10x!HLFHE.eint<6>> to tensor<1x5x!HLFHE.eint<6>> return %r : - tensor<1x5x!HLFHE.eint<6>> + %r = tensor.extract_slice %t[1, 5][1, 5][1, 1] : tensor<2x10x!HLFHE.eint<6>> to tensor<1x5x!HLFHE.eint<6>> + return %r : tensor<1x5x!HLFHE.eint<6>> } )XXX"); @@ -93,9 +92,8 @@ TEST(End2EndJit_EncryptedTensor_2D, extract_slice_stride) { mlir::zamalang::JitCompilerEngine::Lambda lambda = checkedJit(R"XXX( func @main(%t: tensor<2x10x!HLFHE.eint<6>>) -> tensor<1x5x!HLFHE.eint<6>> { - %r = tensor.extract_slice %t[1, 0][1, 5][1, 2] : - tensor<2x10x!HLFHE.eint<6>> to tensor<1x5x!HLFHE.eint<6>> return %r : - tensor<1x5x!HLFHE.eint<6>> + %r = tensor.extract_slice %t[1, 0][1, 5][1, 2] : tensor<2x10x!HLFHE.eint<6>> to tensor<1x5x!HLFHE.eint<6>> + return %r : tensor<1x5x!HLFHE.eint<6>> } )XXX"); @@ -122,9 +120,8 @@ TEST(End2EndJit_EncryptedTensor_2D, insert_slice) { func @main(%t0: tensor<2x10x!HLFHE.eint<6>>, %t1: tensor<2x2x!HLFHE.eint<6>>) -> tensor<2x10x!HLFHE.eint<6>> { - %r = tensor.insert_slice %t1 into %t0[0, 5][2, 2][1, 1] : - tensor<2x2x!HLFHE.eint<6>> into tensor<2x10x!HLFHE.eint<6>> return %r : - tensor<2x10x!HLFHE.eint<6>> + %r = tensor.insert_slice %t1 into %t0[0, 5][2, 2][1, 1] : tensor<2x2x!HLFHE.eint<6>> into tensor<2x10x!HLFHE.eint<6>> + return %r : tensor<2x10x!HLFHE.eint<6>> } )XXX"); @@ -158,4 +155,4 @@ func @main(%t0: tensor<2x10x!HLFHE.eint<6>>, %t1: tensor<2x2x!HLFHE.eint<6>>) } } } -} \ No newline at end of file +}