fix ci error

Signed-off-by: Bangtian Liu <[email protected]>
iree-org · Jan 23, 2025 · 8c5ff94 · 8c5ff94
1 parent 74cb915
commit 8c5ff94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td b/compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.td
@@ -417,7 +417,7 @@ def IREEGPU_TargetChipAttr : AttrDef<IREEGPU_Dialect, "TargetChip"> {
     "uint32_t":$wgp_count,
 
     // An optional SKU identifier to distinguish different models.
-    "std::optional<StringAttr>":$sku,
+    OptionalParameter<"StringAttr">:$sku,
     // An optional extra dict
     // This field allows to inject more features/limits not supported in the
     // above list for better flexibility.

diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp
@@ -118,10 +118,8 @@ TargetAttr createTargetAttr(const TargetDetails &details, StringRef arch,
 
   TargetChipAttr targetChip;
   if (details.chip) {
-    std::optional<StringAttr> skuAttr = std::nullopt;
-    if (details.chip->sku) {
-      skuAttr = StringAttr::get(context, *details.chip->sku);
-    }
+    StringAttr skuAttr =
+        StringAttr::get(context, details.chip->sku.value_or(""));
     targetChip = TargetChipAttr::get(context, details.chip->wgpCount, skuAttr,
                                      DictionaryAttr{});
   }