[SYCLomatic] Add migration support for compound OR assignment operato…

…r for CUDA texture (#2617)
oneapi-src · Jan 16, 2025 · be925b2 · be925b2
1 parent 7bc59b6
commit be925b2
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/clang/lib/DPCT/RulesLang/RulesLangTexture.cpp b/clang/lib/DPCT/RulesLang/RulesLangTexture.cpp
@@ -733,7 +733,8 @@ const Expr *TextureRule::getAssignedBO(const Expr *E, ASTContext &Context) {
   } else if (auto BO = dyn_cast<BinaryOperator>(E)) {
     // If E is BinaryOperator, return E only when it is assign expression,
     // otherwise return nullptr.
-    if (BO->getOpcode() == BO_Assign)
+    auto Opcode = BO->getOpcode();
+    if (Opcode == BO_Assign || Opcode == BO_OrAssign)
       return BO;
   } else if (auto COCE = dyn_cast<CXXOperatorCallExpr>(E)) {
     if (COCE->getOperator() == OO_Equal) {

diff --git a/clang/test/dpct/texture_object_driver.cu b/clang/test/dpct/texture_object_driver.cu
@@ -258,4 +258,36 @@ REDEF_CUDA_RESOURCE_DESC check_typedefed_cuda_type(CUarray arr) {
   rdesc.res.array.hArray = arr;
   return rdesc;
 }
+
+enum TexSurfaceAttrType {
+  UnpackIntegersAsNormalizedFloats,
+  NormalizedAddressing,
+  PerformSRGBToLinearConversion
+};
+
+// CHECK: void checkOrAssign(TexSurfaceAttrType setting) {
+// CHECK-NEXT:   dpct::sampling_info texDesc;
+// CHECK-NEXT:   if (setting == UnpackIntegersAsNormalizedFloats)
+// CHECK-NEXT:     texDesc.set(sycl::coordinate_normalization_mode::unnormalized);
+// CHECK-NEXT:   if (setting == NormalizedAddressing)
+// CHECK-NEXT:     /*
+// CHECK-NEXT:     DPCT1074:{{[0-9]+}}: The SYCL Image class does not support some of the flags used in the original code. Unsupported flags were ignored. Data read from SYCL Image could not be normalized as specified in the original code.
+// CHECK-NEXT:     */
+// CHECK-NEXT:     texDesc.set(sycl::coordinate_normalization_mode::normalized);
+// CHECK-NEXT:   if (setting == PerformSRGBToLinearConversion)
+// CHECK-NEXT:     /*
+// CHECK-NEXT:     DPCT1074:{{[0-9]+}}: The SYCL Image class does not support some of the flags used in the original code. Unsupported flags were ignored. Data read from SYCL Image could not be normalized as specified in the original code.
+// CHECK-NEXT:     */
+// CHECK-NEXT:     texDesc.set(sycl::coordinate_normalization_mode::unnormalized);
+// CHECK-NEXT: }
+void checkOrAssign(TexSurfaceAttrType setting) {
+  CUDA_TEXTURE_DESC texDesc;
+  if (setting == UnpackIntegersAsNormalizedFloats)
+    texDesc.flags |= CU_TRSF_READ_AS_INTEGER;
+  if (setting == NormalizedAddressing)
+    texDesc.flags |= CU_TRSF_NORMALIZED_COORDINATES;
+  if (setting == PerformSRGBToLinearConversion)
+    texDesc.flags |= CU_TRSF_SRGB;
+}
+
 #undef REDEF_CUDA_RESOURCE_DESC