[tuner]: format the code based on comments

Signed-off-by: Bangtian Liu <[email protected]>
iree-org · Dec 7, 2024 · 0f98992 · 0f98992
1 parent 00e8665
commit 0f98992
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 11 deletions.
diff --git a/compiler/bindings/python/IREECompilerDialectsModule.cpp b/compiler/bindings/python/IREECompilerDialectsModule.cpp
@@ -376,16 +376,15 @@ PYBIND11_MODULE(_ireeCompilerDialects, m) {
                 ireeGPULoweringConfigAttrGetSubgroupCount(self);
             MlirAttribute mCountAttr = info.subgroupMCountAttr;
             MlirAttribute nCountAttr = info.subgroupNCountAttr;
-            std::optional<int64_t> mCount =
-                mlirAttributeIsNull(mCountAttr)
-                    ? std::nullopt
-                    : std::optional<int64_t>(
-                          mlirIntegerAttrGetValueInt(mCountAttr));
-            std::optional<int64_t> nCount =
-                mlirAttributeIsNull(nCountAttr)
-                    ? std::nullopt
-                    : std::optional<int64_t>(
-                          mlirIntegerAttrGetValueInt(nCountAttr));
+            std::optional<int64_t> mCount;
+            if (!mlirAttributeIsNull(mCountAttr)) {
+              mCount = mlirIntegerAttrGetValueInt(mCountAttr);
+            }
+
+            std::optional<int64_t> nCount;
+            if (!mlirAttributeIsNull(nCountAttr)) {
+              nCount = mlirIntegerAttrGetValueInt(nCountAttr);
+            }
             return py::make_tuple(mCount, nCount);
           })
       .def_property_readonly(

diff --git a/compiler/src/iree/compiler/API/Internal/IREEGPUDialectCAPI.cpp b/compiler/src/iree/compiler/API/Internal/IREEGPUDialectCAPI.cpp
@@ -229,7 +229,7 @@ ireeGPUTileSizes ireeGPULoweringConfigAttrGetTileSizes(MlirAttribute attr) {
 
   llvm::SmallVector<int64_t> reductions =
       loweringConfigAttr.getStaticTilingLevelSizes(
-          static_cast<int64_t>(
+          llvm::to_underlying(
               mlir::iree_compiler::IREE::GPU::TilingLevel::Reduction),
           nullptr);
   tilesizes.reductionTileSizes = reductions.data();