diff options
| author | Mehdi Amini <joker.eph@gmail.com> | 2025-05-07 13:43:55 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-05-07 13:43:55 +0200 |
| commit | c02aa91939d174a1efda934706d7b523b2fb7e31 (patch) | |
| tree | 1e8160117c4015b201c10142269d03e11f8a71f1 /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | |
| parent | 2f877c2722e882fe6aaaab44d25b7a49ba0612e1 (diff) | |
Revert "[mlir][MemRef] Remove integer address space builders" (#138853)
Reverts llvm/llvm-project#138579
An integration test is broken on the mlir-nvidia* bots.
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
| -rw-r--r-- | mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp index b73c40adcffa..eda2594fbc7c 100644 --- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp +++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp @@ -518,17 +518,15 @@ struct TestVectorScanLowering static Value allocateGlobalSharedMemory(Location loc, OpBuilder &builder, gpu::WarpExecuteOnLane0Op warpOp, Type type) { - Attribute sharedMemorySpaceAttr = - builder.getAttr<gpu::AddressSpaceAttr>(gpu::AddressSpace::Workgroup); + static constexpr int64_t kSharedMemorySpace = 3; // Compute type of shared memory buffer. MemRefType memrefType; if (auto vectorType = dyn_cast<VectorType>(type)) { memrefType = - MemRefType::get(vectorType.getShape(), vectorType.getElementType(), - MemRefLayoutAttrInterface{}, sharedMemorySpaceAttr); + MemRefType::get(vectorType.getShape(), vectorType.getElementType(), {}, + kSharedMemorySpace); } else { - memrefType = MemRefType::get({1}, type, MemRefLayoutAttrInterface{}, - sharedMemorySpaceAttr); + memrefType = MemRefType::get({1}, type, {}, kSharedMemorySpace); } // Get symbol table holding all shared memory globals. |