diff --git a/clang/lib/DPCT/AnalysisInfo.cpp b/clang/lib/DPCT/AnalysisInfo.cpp index d041f24e170d..1623e4faf174 100644 --- a/clang/lib/DPCT/AnalysisInfo.cpp +++ b/clang/lib/DPCT/AnalysisInfo.cpp @@ -5411,9 +5411,7 @@ void KernelCallExpr::printParallelFor(KernelPrinter &Printer, bool IsInSubmit) { Printer.line("[=](" + MapNames::getClNamespace() + "nd_item<1> ", getItemName(), ")", ExecutionConfig.SubGroupSize, " {"); } else { - DpctGlobalInfo::printCtadClass(Printer.indent(), - MapNames::getClNamespace() + "nd_range", 3) - << "("; + Printer << MapNames::getClNamespace() + "nd_range<3>("; if (ExecutionConfig.GroupSize == CanIgnoreRangeStr3D) { Printer << ExecutionConfig.LocalSize; } else if (ExecutionConfig.LocalSize == CanIgnoreRangeStr3D) { diff --git a/clang/test/dpct/ctad.cu b/clang/test/dpct/ctad.cu index c5b1abd2ff61..ab10c0688020 100644 --- a/clang/test/dpct/ctad.cu +++ b/clang/test/dpct/ctad.cu @@ -92,7 +92,7 @@ int main() { // CHECK-NEXT: sycl::local_accessor k_acc_ct1(sycl::range(32), cgh); // CHECK-EMPTY: // CHECK-NEXT: cgh.parallel_for>( - // CHECK-NEXT: sycl::nd_range(sycl::range(1, 1, 1), sycl::range(1, 1, 1)), + // CHECK-NEXT: sycl::nd_range<3>(sycl::range(1, 1, 1), sycl::range(1, 1, 1)), // CHECK-NEXT: [=](sycl::nd_item<3> item_ct1) { // CHECK-NEXT: kernel(1, k_acc_ct1.get_multi_ptr().get()); // CHECK-NEXT: }); @@ -103,12 +103,24 @@ int main() { // CHECK-NEXT: sycl::local_accessor k_acc_ct1(sycl::range(32), cgh); // CHECK-EMPTY: // CHECK-NEXT: cgh.parallel_for>( - // CHECK-NEXT: sycl::nd_range(sycl::range(1, 1, NUM) * sycl::range(1, 1, NUM), sycl::range(1, 1, NUM)), + // CHECK-NEXT: sycl::nd_range<3>(sycl::range(1, 1, NUM) * sycl::range(1, 1, NUM), sycl::range(1, 1, NUM)), // CHECK-NEXT: [=](sycl::nd_item<3> item_ct1) { // CHECK-NEXT: kernel(1, k_acc_ct1.get_multi_ptr().get()); // CHECK-NEXT: }); // CHECK-NEXT: }); kernel<<>>(1); + + // CHECK: q_ct1.submit( + // CHECK-NEXT: [&](sycl::handler &cgh) { + // CHECK-NEXT: sycl::local_accessor k_acc_ct1(sycl::range(32), cgh); + // CHECK-EMPTY: + // CHECK-NEXT: cgh.parallel_for>( + // CHECK-NEXT: sycl::nd_range<3>(deflt * deflt, deflt), + // CHECK-NEXT: [=](sycl::nd_item<3> item_ct1) { + // CHECK-NEXT: kernel(1, k_acc_ct1.get_multi_ptr().get()); + // CHECK-NEXT: }); + // CHECK-NEXT: }); + kernel<<>>(1); }