Skip to content

Commit 54373e0

Browse files
authored
[NFC][mlir][gpu] Make sym_name an inherent attr in GPUModuleOp (#94918)
Make `sym_name` an inherent attr in GPUModuleOp so that it doesn't show in the discardable attributes. The change is safe as the attribute is always expected to be present.
1 parent 77116bd commit 54373e0

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

mlir/include/mlir/Dialect/GPU/IR/GPUOps.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1241,7 +1241,7 @@ def GPU_BarrierOp : GPU_Op<"barrier"> {
12411241
def GPU_GPUModuleOp : GPU_Op<"module", [
12421242
DataLayoutOpInterface, HasDefaultDLTIDataLayout, IsolatedFromAbove,
12431243
SymbolTable, Symbol, SingleBlockImplicitTerminator<"ModuleEndOp">
1244-
]>, Arguments<(ins
1244+
]>, Arguments<(ins SymbolNameAttr:$sym_name,
12451245
OptionalAttr<GPUNonEmptyTargetArrayAttr>:$targets,
12461246
OptionalAttr<OffloadingTranslationAttr>:$offloadingHandler)> {
12471247
let summary = "A top level compilation unit containing code to be run on a GPU.";

mlir/lib/Dialect/GPU/IR/GPUDialect.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1730,12 +1730,11 @@ void GPUModuleOp::build(OpBuilder &builder, OperationState &result,
17301730
StringRef name, ArrayAttr targets,
17311731
Attribute offloadingHandler) {
17321732
ensureTerminator(*result.addRegion(), builder, result.location);
1733-
result.attributes.push_back(builder.getNamedAttr(
1734-
::mlir::SymbolTable::getSymbolAttrName(), builder.getStringAttr(name)));
17351733

17361734
Properties &props = result.getOrAddProperties<Properties>();
17371735
if (targets)
17381736
props.targets = targets;
1737+
props.setSymName(builder.getStringAttr(name));
17391738
props.offloadingHandler = offloadingHandler;
17401739
}
17411740

@@ -1751,11 +1750,11 @@ ParseResult GPUModuleOp::parse(OpAsmParser &parser, OperationState &result) {
17511750
StringAttr nameAttr;
17521751
ArrayAttr targetsAttr;
17531752

1754-
if (parser.parseSymbolName(nameAttr, mlir::SymbolTable::getSymbolAttrName(),
1755-
result.attributes))
1753+
if (parser.parseSymbolName(nameAttr))
17561754
return failure();
17571755

17581756
Properties &props = result.getOrAddProperties<Properties>();
1757+
props.setSymName(nameAttr);
17591758

17601759
// Parse the optional offloadingHandler
17611760
if (succeeded(parser.parseOptionalLess())) {

0 commit comments

Comments
 (0)