Skip to content

Commit 3e5e48a

Browse files
authored
1 parent ce6369e commit 3e5e48a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

flang/lib/Optimizer/Transforms/CufOpConversion.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,18 +370,20 @@ struct CufDataTransferOpConversion
370370
mlir::Type srcTy = fir::unwrapRefType(op.getSrc().getType());
371371
mlir::Type dstTy = fir::unwrapRefType(op.getDst().getType());
372372

373-
unsigned mode;
373+
mlir::Location loc = op.getLoc();
374+
unsigned mode = 0;
374375
if (op.getTransferKind() == cuf::DataTransferKind::HostDevice) {
375376
mode = kHostToDevice;
376377
} else if (op.getTransferKind() == cuf::DataTransferKind::DeviceHost) {
377378
mode = kDeviceToHost;
378379
} else if (op.getTransferKind() == cuf::DataTransferKind::DeviceDevice) {
379380
mode = kDeviceToDevice;
381+
} else {
382+
mlir::emitError(loc, "unsupported transfer kind\n");
380383
}
381384

382385
auto mod = op->getParentOfType<mlir::ModuleOp>();
383386
fir::FirOpBuilder builder(rewriter, mod);
384-
mlir::Location loc = op.getLoc();
385387
fir::KindMapping kindMap{fir::getKindMapping(mod)};
386388
mlir::Value modeValue =
387389
builder.createIntegerConstant(loc, builder.getI32Type(), mode);

0 commit comments

Comments
 (0)