Skip to content

Commit 2807dd1

Browse files
committed
[mlir][memref][spirv]: Address Feedback
* Remove use of initializer `if` statement to aid readability. * Remove switch case macro and make statements explicit. Signed-off-by: Jack Frankland <[email protected]>
1 parent cd1106d commit 2807dd1

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

mlir/lib/Conversion/MemRefToSPIRV/MemRefToSPIRV.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -715,9 +715,9 @@ ImageLoadOpPattern::matchAndRewrite(memref::LoadOp loadOp, OpAdaptor adaptor,
715715
// for plain images and the OpImageRead instruction needs to be materialized
716716
// instead or texels need to be accessed via atomics through a texel pointer.
717717
// Future work will generalize support to plain images.
718-
if (auto convertedPointeeType = cast<spirv::PointerType>(
719-
getTypeConverter()->convertType(loadOp.getMemRefType()));
720-
!isa<spirv::SampledImageType>(convertedPointeeType.getPointeeType()))
718+
auto convertedPointeeType = cast<spirv::PointerType>(
719+
getTypeConverter()->convertType(loadOp.getMemRefType()));
720+
if (!isa<spirv::SampledImageType>(convertedPointeeType.getPointeeType()))
721721
return rewriter.notifyMatchFailure(loadOp,
722722
"cannot lower memrefs which do not "
723723
"convert to SPIR-V sampled images");

mlir/lib/Dialect/SPIRV/Transforms/SPIRVConversion.cpp

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -588,17 +588,16 @@ static Type convertMemrefType(const spirv::TargetEnv &targetEnv,
588588
}
589589

590590
const auto dim = [rank]() {
591-
#define DIM_CASE(DIM) \
592-
case DIM: \
593-
return spirv::Dim::Dim##DIM##D
594591
switch (rank) {
595-
DIM_CASE(1);
596-
DIM_CASE(2);
597-
DIM_CASE(3);
592+
case 1:
593+
return spirv::Dim::Dim1D;
594+
case 2:
595+
return spirv::Dim::Dim2D;
596+
case 3:
597+
return spirv::Dim::Dim3D;
598598
default:
599599
llvm_unreachable("Invalid memref rank!");
600600
}
601-
#undef DIM_CASE
602601
}();
603602

604603
// Note that we currently only support lowering to single element texels

0 commit comments

Comments
 (0)