Fix build issue on MSVC by not having a conditional on disjoint types.

llvm · Sep 7, 2023 · a8fd275 · a8fd275
1 parent 67ae97f
commit a8fd275
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp b/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp
@@ -1273,13 +1273,13 @@ class ConvertAtenScatterReduceTwoOp
         // Set the values in the input tensor to the smallest element of that
         // type
         TypedAttr minAttr = getNumericLimit(rewriter, srcType.getElementType(),
-                                       /*getMin=*/true);
+                                            /*getMin=*/true);
         normalizationValue = rewriter.create<arith::ConstantOp>(loc, minAttr);
       } else if (reduceEnum == torch_upstream::ReductionType::MIN) {
         // Set the values in the input tensor to the largest element of that
         // type
         TypedAttr maxAttr = getNumericLimit(rewriter, srcType.getElementType(),
-                                       /*getMin=*/false);
+                                            /*getMin=*/false);
         normalizationValue = rewriter.create<arith::ConstantOp>(loc, maxAttr);
       }
 
@@ -1551,10 +1551,10 @@ class ConvertAtenCumsumOp : public OpConversionPattern<AtenCumsumOp> {
     Value result = createTMTensorScanOp(
         rewriter, loc, input, output, acc, dim, /*inclusive=*/true,
         [](OpBuilder &b, Location loc, Value input, Value acc) {
-          Value sum = (input.getType().isa<mlir::FloatType>()
-                           ? b.create<arith::AddFOp>(loc, input, acc)
-                           : b.create<arith::AddIOp>(loc, input, acc))
-                          ->getResult(0);
+          Value sum =
+              (input.getType().isa<mlir::FloatType>()
+                   ? b.create<arith::AddFOp>(loc, input, acc)->getResult(0)
+                   : b.create<arith::AddIOp>(loc, input, acc)->getResult(0));
           b.create<TMTensor::YieldOp>(loc, sum);
         });