Merge pull request #2250 from yingcong-wu/yc-PR/1028-leak-check-option

[DevAsan] Add an option to enable/disable leak detection

Author: kbenzie

source/loader/layers/sanitizer/asan/asan_interceptor.cpp

@@ -932,7 +932,8 @@ ContextInfo::~ContextInfo() {
     assert(Result == UR_RESULT_SUCCESS);
 
     // check memory leaks
-    if (getAsanInterceptor()->isNormalExit()) {
+    if (getAsanInterceptor()->getOptions().DetectLeaks &&
+        getAsanInterceptor()->isNormalExit()) {
         std::vector<AllocationIterator> AllocInfos =
             getAsanInterceptor()->findAllocInfoByContext(Handle);
         for (const auto &It : AllocInfos) {

source/loader/layers/sanitizer/asan/asan_interceptor.hpp

@@ -310,6 +310,8 @@ class AsanInterceptor {
     ur_result_t registerSpirKernels(ur_program_handle_t Program);
 
   private:
+    // m_Options may be used in other places, place it at the top
+    AsanOptions m_Options;
     std::unordered_map<ur_context_handle_t, std::shared_ptr<ContextInfo>>
         m_ContextMap;
     ur_shared_mutex m_ContextMapMutex;
@@ -335,8 +337,6 @@ class AsanInterceptor {
 
     std::unique_ptr<Quarantine> m_Quarantine;
 
-    AsanOptions m_Options;
-
     std::unordered_set<ur_adapter_handle_t> m_Adapters;
     ur_shared_mutex m_AdaptersMutex;
 

source/loader/layers/sanitizer/asan/asan_options.cpp

@@ -88,6 +88,7 @@ AsanOptions::AsanOptions() {
     SetBoolOption("detect_locals", DetectLocals);
     SetBoolOption("detect_privates", DetectPrivates);
     SetBoolOption("print_stats", PrintStats);
+    SetBoolOption("detect_leaks", DetectLeaks);
 
     auto KV = OptionsEnvMap->find("quarantine_size_mb");
     if (KV != OptionsEnvMap->end()) {

source/loader/layers/sanitizer/asan/asan_options.hpp

@@ -26,6 +26,7 @@ struct AsanOptions {
     bool DetectPrivates = true;
     bool PrintStats = false;
     bool DetectKernelArguments = true;
+    bool DetectLeaks = true;
 
     explicit AsanOptions();
 };