Index: lib/CodeGen/CGStmtOpenMP.cpp =================================================================== --- lib/CodeGen/CGStmtOpenMP.cpp +++ lib/CodeGen/CGStmtOpenMP.cpp @@ -921,7 +921,13 @@ break; } } + + const auto *CS = + S.getAssociatedStmt()->IgnoreContainers(/*IgnoreCaptured=*/true); + if (const auto *EWC = dyn_cast(CS)) + enterFullExpression(EWC); InlinedOpenMPRegionScopeRAII Region(*this, S); + EmitOMPAtomicExpr(*this, Kind, IsSeqCst, S.getX(), S.getV(), S.getExpr(), S.getLocStart()); } Index: lib/Sema/SemaOpenMP.cpp =================================================================== --- lib/Sema/SemaOpenMP.cpp +++ lib/Sema/SemaOpenMP.cpp @@ -3267,6 +3267,9 @@ } auto Body = CS->getCapturedStmt(); + if (auto *EWC = dyn_cast(Body)) + Body = EWC->getSubExpr(); + Expr *X = nullptr; Expr *V = nullptr; Expr *E = nullptr; Index: test/OpenMP/atomic_codegen.cpp =================================================================== --- test/OpenMP/atomic_codegen.cpp +++ test/OpenMP/atomic_codegen.cpp @@ -1,7 +1,37 @@ +// RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp=libiomp5 -fexceptions -fcxx-exceptions -x c++ -emit-llvm %s -o - | FileCheck %s // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp=libiomp5 -fexceptions -fcxx-exceptions -gline-tables-only -x c++ -emit-llvm %s -o - | FileCheck %s --check-prefix=TERM_DEBUG // expected-no-diagnostics int a; +int b; + +struct St { + St() {} + ~St() {} + int &get() { return a; } +}; + +// CHECK-LABEL: parallel_atomic_ewc +void parallel_atomic_ewc() { +#pragma omp parallel + { + // CHECK: invoke void @_ZN2StC1Ev(%struct.St* [[TEMP_ST_ADDR:%.+]]) + // CHECK: [[SCALAR_ADDR:%.+]] = invoke dereferenceable(4) i32* @_ZN2St3getEv(%struct.St* [[TEMP_ST_ADDR]]) + // CHECK: [[SCALAR_VAL:%.+]] = load atomic i32, i32* [[SCALAR_ADDR]] monotonic + // CHECK: store i32 [[SCALAR_VAL]], i32* @b + // CHECK: invoke void @_ZN2StD1Ev(%struct.St* [[TEMP_ST_ADDR]]) +#pragma omp atomic read + b = St().get(); + // CHECK: invoke void @_ZN2StC1Ev(%struct.St* [[TEMP_ST_ADDR:%.+]]) + // CHECK: [[SCALAR_ADDR:%.+]] = invoke dereferenceable(4) i32* @_ZN2St3getEv(%struct.St* [[TEMP_ST_ADDR]]) + // CHECK: [[B_VAL:%.+]] = load i32, i32* @b + // CHECK: store atomic i32 [[B_VAL]], i32* [[SCALAR_ADDR]] monotonic + // CHECK: invoke void @_ZN2StD1Ev(%struct.St* [[TEMP_ST_ADDR]]) +#pragma omp atomic write + St().get() = b; + } +} + int &foo() { return a; } // TERM_DEBUG-LABEL: parallel_atomic @@ -12,7 +42,7 @@ // TERM_DEBUG-NOT: __kmpc_global_thread_num // TERM_DEBUG: invoke {{.*}}foo{{.*}}() // TERM_DEBUG: unwind label %[[TERM_LPAD:.+]], - // TERM_DEBUG: load atomic i32, i32* @{{.+}} monotonic, {{.*}}!dbg [[READ_LOC:![0-9]+]] + // TERM_DEBUG: load atomic i32, i32* @{{.+}} monotonic, {{.*}}!dbg [[READ_LOC:![0-9]+]] foo() = a; #pragma omp atomic write // TERM_DEBUG-NOT: __kmpc_global_thread_num @@ -26,5 +56,5 @@ a = foo(); } } -// TERM_DEBUG-DAG: [[READ_LOC]] = !MDLocation(line: 11, -// TERM_DEBUG-DAG: [[WRITE_LOC]] = !MDLocation(line: 17, +// TERM_DEBUG-DAG: [[READ_LOC]] = !MDLocation(line: 41, +// TERM_DEBUG-DAG: [[WRITE_LOC]] = !MDLocation(line: 47,