Create extenstion -D options for clang based on active extension string.

Change-Id: Iee5aa0d0259abc3ab821396b4c8e7b0366161d78

Author: PawelJurek

IGC/OCLFE/igd_fcl_mcl/source/clang_tb.cpp

@@ -1037,6 +1037,43 @@ namespace TC
         return internalDefines;
     }
 
+    // The expected extensions input string is in a form:
+    // -cl-ext=-all,+supported_ext_name,+second_supported_ext_name
+    std::string GetCDefinesForExtensions(llvm::StringRef extensions, unsigned int oclStd) {
+
+      std::string extDefines;
+
+      // check for the last occurence of -all, as it invalidates previous occurances.
+      const StringRef clExtPrefix = "-cl-ext=-all,";
+      size_t pos = extensions.rfind(clExtPrefix);
+      if (pos == llvm::StringRef::npos) {
+        // If this string does not exist the input string does not contain valid extension list
+        // or it has all extensions disabled (-all without colon afterwards).
+        return extDefines;
+      }
+
+      extensions = extensions.substr(pos+clExtPrefix.size());
+
+      // string with defines should be similar in size, ",+" will change to "-D".
+      extDefines.reserve(extensions.size());
+
+      llvm::SmallVector<StringRef, 0> v;
+      extensions.split(v, ',');
+
+      for (auto ext : v) {
+        if (ext.consume_front("+")) {
+          if (ext.equals("cl_intel_device_side_avc_motion_estimation")) {
+            // If the user provided -cl-std option we need to add the define only if it's 1.2 and above.
+            // This is because clang will not allow declarations of extension's functions which use avc types otherwise.
+            if (!(oclStd >= 120 || oclStd == 0)) continue;
+          }
+          extDefines.append(" -D").append(ext);
+        }
+      }
+
+      return extDefines;
+    }
+
     /*****************************************************************************\
 
     Function:
@@ -1407,47 +1444,10 @@ namespace TC
 
         optionsEx += " " + extensions;
 
+        unsigned int oclStd = GetOclCVersionFromOptions(pInputArgs->options.data(), nullptr, pInputArgs->oclVersion, exceptString);
         // get additional -D flags from internal options
         optionsEx += " " + GetCDefinesFromInternalOptions(pInternalOptions);
-
-        if (extensions.find("cl_intel_subgroups_short") != std::string::npos)
-        {
-          optionsEx += " -Dcl_intel_subgroups_short";
-        }
-        if (extensions.find("cl_intel_subgroups_char") != std::string::npos)
-        {
-            optionsEx += " -Dcl_intel_subgroups_char";
-        }
-        if (extensions.find("cl_intel_subgroups_long") != std::string::npos)
-        {
-            optionsEx += " -Dcl_intel_subgroups_long";
-        }
-        if (extensions.find("cl_intel_media_block_io") != std::string::npos)
-        {
-          optionsEx += " -Dcl_intel_media_block_io";
-        }
-        if (extensions.find("cl_intel_device_side_avc_motion_estimation") != std::string::npos)
-        {
-          // If the user provided -cl-std option we need to add the define only if it's 1.2 and above.
-          // This is because clang will not allow declarations of extension's functions which use avc types otherwise.
-          unsigned int oclStd = GetOclCVersionFromOptions(pInputArgs->options.data(), nullptr, pInputArgs->oclVersion, exceptString);
-          if (oclStd >= 120 || oclStd == 0) {
-            optionsEx += " -Dcl_intel_device_side_avc_motion_estimation";
-          }
-        }
-        if (extensions.find("cl_intel_64bit_global_atomics_placeholder") != std::string::npos)
-        {
-          optionsEx += " -Dcl_intel_64bit_global_atomics_placeholder";
-        }
-        if (extensions.find("cl_khr_int64_base_atomics") != std::string::npos)
-        {
-          optionsEx += " -Dcl_khr_int64_base_atomics";
-        }
-        if (extensions.find("cl_khr_int64_extended_atomics") != std::string::npos)
-        {
-          optionsEx += " -Dcl_khr_int64_extended_atomics";
-        }
-
+        optionsEx += " " + GetCDefinesForExtensions(extensions, oclStd);
         optionsEx += " -D__IMAGE_SUPPORT__ -D__ENDIAN_LITTLE__";
 
         IOCLFEBinaryResult *pResultPtr = NULL;