llvm-project.git/mlir/lib/Dialect/GPU/Transforms/ROCDLAttachTarget.cpp, branch users/shawbyoung/spr/main.boltnfc-refactoring-callgraph

[mlir] Use llvm::unique (NFC) (#96415)

2024-06-24T18:54:02+00:00

[mlir] Apply ClangTidy fixes (NFC)

2023-08-23T15:18:59+00:00

Prefer to use .empty() instead of checking size().

[mlir] Disentangle dialect and extension registrations.

2023-08-22T00:40:09+00:00

This revision avoids the registration of dialect extensions in Pass::getDependentDialects.

Such registration of extensions can be dangerous because `DialectRegistry::isSubsetOf` is
always guaranteed to return false for extensions (i.e. there is no mechanism to track
whether a lambda is already in the list of already registered extensions).
When the context is already in a multi-threaded mode, this is guaranteed to assert.

Arguably a more structured registration mechanism for extensions with a unique ExtensionID
could be envisioned in the future.

In the process of cleaning this up, multiple usage inconsistencies surfaced around the
registration of translation extensions that this revision also cleans up.

Reviewed By: springerm

Differential Revision: https://reviews.llvm.org/D157703

[mlir][gpu] Add passes to attach (NVVM|ROCDL) target attributes to GPU Modules

2023-08-12T00:45:26+00:00

Adds the passes `nvvm-attach-target` & `rocdl-attach-target for attaching `nvvm.target` & `rocdl.target` attributes to GPU Modules.

These passes search GPU Modules in the immediate region of the Op being acted on, attaching the target attribute to the module.
Modules can be selected using a regex string, allowing fine grain attachment of targets, see the test `attach-target.mlir` for an example.

Depends on D154153

Reviewed By: mehdi_amini

Differential Revision: https://reviews.llvm.org/D157351