diff --git a/flang/lib/Lower/Bridge.cpp b/flang/lib/Lower/Bridge.cpp index 6b7efe6b57db3..a5f0beff9ec22 100644 --- a/flang/lib/Lower/Bridge.cpp +++ b/flang/lib/Lower/Bridge.cpp @@ -4898,7 +4898,6 @@ class FirConverter : public Fortran::lower::AbstractConverter { // device = device if (lhsIsDevice && rhsIsDevice) { - assert(rhs.isVariable() && "CUDA Fortran assignment rhs is not legal"); auto transferKindAttr = cuf::DataTransferKindAttr::get( builder.getContext(), cuf::DataTransferKind::DeviceDevice); cuf::DataTransferOp::create(builder, loc, rhsVal, lhsVal, shape, diff --git a/flang/test/Lower/CUDA/cuda-data-transfer.cuf b/flang/test/Lower/CUDA/cuda-data-transfer.cuf index 3a4aff977b7a5..9abf19b778690 100644 --- a/flang/test/Lower/CUDA/cuda-data-transfer.cuf +++ b/flang/test/Lower/CUDA/cuda-data-transfer.cuf @@ -444,3 +444,15 @@ subroutine sub23(n) end subroutine ! CHECK-LABEL: func.func @_QPsub23 + +subroutine sub24() + real, managed :: m + real, device :: d(4) + m = d(1) +end + +! CHECK-LABEL: func.func @_QPsub24() +! CHECK: %[[D:.*]]:2 = hlfir.declare %1(%2) {data_attr = #cuf.cuda, uniq_name = "_QFsub24Ed"} : (!fir.ref>, !fir.shape<1>) -> (!fir.ref>, !fir.ref>) +! CHECK: %[[M:.*]]:2 = hlfir.declare %4 {data_attr = #cuf.cuda, uniq_name = "_QFsub24Em"} : (!fir.ref) -> (!fir.ref, !fir.ref) +! CHECK: %[[D1:.*]] = hlfir.designate %[[D]]#0 (%c1{{.*}}) : (!fir.ref>, index) -> !fir.ref +! CHECK: cuf.data_transfer %[[D1]] to %[[M]]#0 {transfer_kind = #cuf.cuda_transfer} : !fir.ref, !fir.ref