-
Notifications
You must be signed in to change notification settings - Fork 608
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Enabling linking in the ROCM compiler target.
This does exactly what the LLVMCPU side does - which is bad for compile time (serializes LLVM codegen) but much better for runtime. Future improvements should move LLVM codegen to the linking phase so it can happen in parallel and then perform the linking using LLVM's linker (each executable turned into a .o and then combined into a .so, or last-level bitcode if then we just want serialization to be bitcode to machine code).
- Loading branch information
Showing
13 changed files
with
356 additions
and
130 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
240 changes: 120 additions & 120 deletions
240
compiler/src/iree/compiler/API/Internal/IREEGPUDialectCAPI.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,120 +1,120 @@ | ||
| // Copyright 2024 The IREE Authors | ||
| // | ||
| // Licensed under the Apache License v2.0 with LLVM Exceptions. | ||
| // See https://llvm.org/LICENSE.txt for license information. | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
|
|
||
| #include "iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.h" | ||
| #include "iree/compiler/dialects/iree_gpu.h" | ||
| #include "mlir-c/IR.h" | ||
| #include "mlir/CAPI/IR.h" | ||
| #include "mlir/CAPI/Support.h" | ||
|
|
||
| bool ireeAttributeIsAGPUPipelineOptionsAttr(MlirAttribute attr) { | ||
| return llvm::isa<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGet(MlirContext mlirCtx, bool *prefetchSharedMemory, | ||
| bool *noReduceSharedMemoryBankConflicts, | ||
| MlirAttribute *reorderWorkgroupsStrategy) { | ||
| mlir::MLIRContext *ctx = unwrap(mlirCtx); | ||
| mlir::Builder b(ctx); | ||
| auto prefetchSharedMemoryAttr = mlir::BoolAttr(); | ||
| if (prefetchSharedMemory) { | ||
| prefetchSharedMemoryAttr = b.getBoolAttr(*prefetchSharedMemory); | ||
| } | ||
| auto noReduceSharedMemoryBankConflictsAttr = mlir::BoolAttr(); | ||
| if (noReduceSharedMemoryBankConflicts) { | ||
| noReduceSharedMemoryBankConflictsAttr = | ||
| b.getBoolAttr(*noReduceSharedMemoryBankConflicts); | ||
| } | ||
| auto strategyAttr = | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr(); | ||
| if (reorderWorkgroupsStrategy) { | ||
| strategyAttr = llvm::dyn_cast< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(*reorderWorkgroupsStrategy)); | ||
| } | ||
| return wrap(mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr::get( | ||
| ctx, prefetchSharedMemoryAttr, noReduceSharedMemoryBankConflictsAttr, | ||
| strategyAttr)); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGetPrefetchSharedMemory(MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getPrefetchSharedMemory()); | ||
| } | ||
|
|
||
| MlirAttribute ireeGPUPipelineOptionsAttrGetNoReduceSharedMemoryBankConflicts( | ||
| MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getNoReduceSharedMemoryBankConflicts()); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGetReorderWorkgroupsStrategy(MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getReorderWorkgroupsStrategy()); | ||
| } | ||
|
|
||
| MlirTypeID ireeGPUPipelineOptionsAttrGetTypeID() { | ||
| return wrap( | ||
| mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr::getTypeID()); | ||
| } | ||
|
|
||
| static_assert( | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumNone) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::None) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumSwizzle) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::Swizzle) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumTranspose) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::Transpose) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumTranspose) == | ||
| mlir::iree_compiler::IREE::GPU:: | ||
| getMaxEnumValForReorderWorkgroupsStrategy(), | ||
| "ireeGPUReorderWorkgroupsStrategyEnum and " | ||
| "mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategy definitions " | ||
| "have diverged"); | ||
|
|
||
| bool ireeAttributeIsAGPUReorderWorkgroupsStrategyAttr(MlirAttribute attr) { | ||
| return llvm::isa< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(attr)); | ||
| } | ||
|
|
||
| MlirTypeID ireeGPUReorderWorkgroupsStrategyAttrGetTypeID() { | ||
| return wrap(mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr:: | ||
| getTypeID()); | ||
| } | ||
|
|
||
| MlirAttribute ireeGPUReorderWorkgroupsStrategyAttrGet( | ||
| MlirContext mlirCtx, ireeGPUReorderWorkgroupsStrategyEnum value) { | ||
| mlir::MLIRContext *ctx = unwrap(mlirCtx); | ||
| return wrap( | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr::get( | ||
| ctx, static_cast< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategy>( | ||
| value))); | ||
| } | ||
|
|
||
| ireeGPUReorderWorkgroupsStrategyEnum | ||
| ireeGPUReorderWorkgroupsStrategyAttrGetValue(MlirAttribute attr) { | ||
| assert(ireeAttributeIsAGPUReorderWorkgroupsStrategyAttr(attr) && | ||
| "attr is not a GPUReorderWorkgroupsStrategyAttr"); | ||
| return static_cast<ireeGPUReorderWorkgroupsStrategyEnum>( | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(attr)) | ||
| .getValue()); | ||
| } | ||
| // Copyright 2024 The IREE Authors | ||
| // | ||
| // Licensed under the Apache License v2.0 with LLVM Exceptions. | ||
| // See https://llvm.org/LICENSE.txt for license information. | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
|
|
||
| #include "iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.h" | ||
| #include "iree/compiler/dialects/iree_gpu.h" | ||
| #include "mlir-c/IR.h" | ||
| #include "mlir/CAPI/IR.h" | ||
| #include "mlir/CAPI/Support.h" | ||
|
|
||
| bool ireeAttributeIsAGPUPipelineOptionsAttr(MlirAttribute attr) { | ||
| return llvm::isa<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGet(MlirContext mlirCtx, bool *prefetchSharedMemory, | ||
| bool *noReduceSharedMemoryBankConflicts, | ||
| MlirAttribute *reorderWorkgroupsStrategy) { | ||
| mlir::MLIRContext *ctx = unwrap(mlirCtx); | ||
| mlir::Builder b(ctx); | ||
| auto prefetchSharedMemoryAttr = mlir::BoolAttr(); | ||
| if (prefetchSharedMemory) { | ||
| prefetchSharedMemoryAttr = b.getBoolAttr(*prefetchSharedMemory); | ||
| } | ||
| auto noReduceSharedMemoryBankConflictsAttr = mlir::BoolAttr(); | ||
| if (noReduceSharedMemoryBankConflicts) { | ||
| noReduceSharedMemoryBankConflictsAttr = | ||
| b.getBoolAttr(*noReduceSharedMemoryBankConflicts); | ||
| } | ||
| auto strategyAttr = | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr(); | ||
| if (reorderWorkgroupsStrategy) { | ||
| strategyAttr = llvm::dyn_cast< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(*reorderWorkgroupsStrategy)); | ||
| } | ||
| return wrap(mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr::get( | ||
| ctx, prefetchSharedMemoryAttr, noReduceSharedMemoryBankConflictsAttr, | ||
| strategyAttr)); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGetPrefetchSharedMemory(MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getPrefetchSharedMemory()); | ||
| } | ||
|
|
||
| MlirAttribute ireeGPUPipelineOptionsAttrGetNoReduceSharedMemoryBankConflicts( | ||
| MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getNoReduceSharedMemoryBankConflicts()); | ||
| } | ||
|
|
||
| MlirAttribute | ||
| ireeGPUPipelineOptionsAttrGetReorderWorkgroupsStrategy(MlirAttribute attr) { | ||
| auto gpuAttr = | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr>( | ||
| unwrap(attr)); | ||
| return wrap(gpuAttr.getReorderWorkgroupsStrategy()); | ||
| } | ||
|
|
||
| MlirTypeID ireeGPUPipelineOptionsAttrGetTypeID() { | ||
| return wrap( | ||
| mlir::iree_compiler::IREE::GPU::GPUPipelineOptionsAttr::getTypeID()); | ||
| } | ||
|
|
||
| static_assert( | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumNone) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::None) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumSwizzle) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::Swizzle) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumTranspose) == | ||
| static_cast<uint32_t>(mlir::iree_compiler::IREE::GPU:: | ||
| ReorderWorkgroupsStrategy::Transpose) && | ||
| static_cast<uint32_t>(ireeGPUReorderWorkgroupsStrategyEnumTranspose) == | ||
| mlir::iree_compiler::IREE::GPU:: | ||
| getMaxEnumValForReorderWorkgroupsStrategy(), | ||
| "ireeGPUReorderWorkgroupsStrategyEnum and " | ||
| "mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategy definitions " | ||
| "have diverged"); | ||
|
|
||
| bool ireeAttributeIsAGPUReorderWorkgroupsStrategyAttr(MlirAttribute attr) { | ||
| return llvm::isa< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(attr)); | ||
| } | ||
|
|
||
| MlirTypeID ireeGPUReorderWorkgroupsStrategyAttrGetTypeID() { | ||
| return wrap(mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr:: | ||
| getTypeID()); | ||
| } | ||
|
|
||
| MlirAttribute ireeGPUReorderWorkgroupsStrategyAttrGet( | ||
| MlirContext mlirCtx, ireeGPUReorderWorkgroupsStrategyEnum value) { | ||
| mlir::MLIRContext *ctx = unwrap(mlirCtx); | ||
| return wrap( | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr::get( | ||
| ctx, static_cast< | ||
| mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategy>( | ||
| value))); | ||
| } | ||
|
|
||
| ireeGPUReorderWorkgroupsStrategyEnum | ||
| ireeGPUReorderWorkgroupsStrategyAttrGetValue(MlirAttribute attr) { | ||
| assert(ireeAttributeIsAGPUReorderWorkgroupsStrategyAttr(attr) && | ||
| "attr is not a GPUReorderWorkgroupsStrategyAttr"); | ||
| return static_cast<ireeGPUReorderWorkgroupsStrategyEnum>( | ||
| llvm::cast<mlir::iree_compiler::IREE::GPU::ReorderWorkgroupsStrategyAttr>( | ||
| unwrap(attr)) | ||
| .getValue()); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.