add CMake option to custom PCH extension map

Author: haonanya1

CMakeLists.txt

@@ -106,6 +106,12 @@ if (NOT DEFINED COMMON_CLANG_LIBRARY_NAME)
 endif()
 set(TARGET_NAME ${COMMON_CLANG_LIBRARY_NAME}${BUILD_PLATFORM} )
 
+option(CUSTOM_PCH_EXTENSION_MAP "Set PCH extension map" OFF)
+if(CUSTOM_PCH_EXTENSION_MAP)
+  add_definitions(-DPCH_EXTENSION="cl_khr_3d_image_writes,cl_khr_depth_images,cl_khr_global_int32_base_atomics,cl_khr_global_int32_extended_atomics,cl_khr_int64_base_atomics,cl_khr_int64_extended_atomics,cl_khr_local_int32_base_atomics,cl_khr_local_int32_extended_atomics,cl_intel_subgroups,cl_intel_subgroups_short")
+endif()
+
+
 if(NOT USE_PREBUILT_LLVM)
 
     if(NOT LLVM_EXTERNAL_CLANG_SOURCE_DIR)

options_compile.cpp

@@ -257,27 +257,36 @@ std::string EffectiveOptionsFilter::processOptions(const OpenCLArgList &args,
   std::map<std::string, bool> extMap{
       {"cl_khr_3d_image_writes", true},
       {"cl_khr_depth_images", true},
-#ifdef GPU_EXT
       {"cl_khr_fp16", true},
-      {"cl_khr_mipmap_image", true},
-      {"cl_khr_mipmap_image_writes", true},
-      {"cl_khr_subgroups", true},
-      {"cl_intel_device_side_avc_motion_estimation", true},
-      {"cl_intel_planar_yuv", true},
 #ifdef _WIN32
       // cl_khr_gl_msaa_sharing is only supported on Windows [NEO].
       {"cl_khr_gl_msaa_sharing", true},
-#endif
 #endif
       {"cl_khr_global_int32_base_atomics", true},
       {"cl_khr_global_int32_extended_atomics", true},
       {"cl_khr_int64_base_atomics", true},
       {"cl_khr_int64_extended_atomics", true},
       {"cl_khr_local_int32_base_atomics", true},
       {"cl_khr_local_int32_extended_atomics", true},
+      {"cl_khr_mipmap_image", true},
+      {"cl_khr_mipmap_image_writes", true},
+      {"cl_khr_subgroups", true},
+      {"cl_intel_device_side_avc_motion_estimation", true},
+      {"cl_intel_planar_yuv", true},
       {"cl_intel_subgroups", true},
       {"cl_intel_subgroups_short", true}};
 
+#ifdef PCH_EXTENSION
+  extMap.clear();
+  std::string customPCHExtStr(PCH_EXTENSION);
+  std::stringstream sstr(customPCHExtStr);
+  while(sstr.good()) {
+    std::string substr;
+    getline(sstr, substr, ',');
+    extMap[substr] = true;
+  }
+#endif
+
   auto parseClExt = [&](const std::string &clExtStr) {
     llvm::StringRef clExtRef(clExtStr);
     clExtRef.consume_front("-cl-ext=");