UCT/ROCM: Fix dmabuf support check for coreos.

Author: hahahannes

AUTHORS

@@ -42,6 +42,7 @@ Gonzalo Brito Gadeschi <gonzalob@nvidia.com>
 Graham Lopez <lopezmg@ornl.gov>
 Guy Ealey Morag <gealeymorag@nvidia.com>
 Guy Shattah <sguy@mellanox.com>
+Hannes Hansen <hannes.jakob.hansen@cern.ch>
 Hessam Mirsadeghi <hmirsadeghi@nvidia.com>
 Hiroyuki Sato <hiroysato@gmail.com>
 Hod Badihi <hbadihi@nvidia.com>

src/uct/rocm/base/rocm_base.c

@@ -310,34 +310,71 @@ int uct_rocm_base_is_dmabuf_supported()
     struct utsname utsname;
     char kernel_conf_file[128];
     char buf[256];
+    const char *fmts[] = {
+        "/boot/config-%s",
+        "/usr/src/linux-%s/.config",
+        "/usr/src/linux/.config",
+        "/usr/lib/modules/%s/config",
+        "/usr/lib/ostree-boot/config-%s",
+        "/usr/lib/kernel/config-%s",
+        "/usr/src/linux-headers-%s/.config",
+        "/lib/modules/%s/build/.config"
+    };
 
     if (uname(&utsname) == -1) {
         ucs_trace("could not get kernel name");
         goto out;
     }
 
-    ucs_snprintf_safe(kernel_conf_file, sizeof(kernel_conf_file),
-                      "/boot/config-%s", utsname.release);
-    fp = fopen(kernel_conf_file, "r");
+    for (size_t i = 0; i < sizeof(fmts) / sizeof(fmts[0]); ++i) {
+        const char *path = fmts[i];
+        ucs_snprintf_safe(kernel_conf_file, sizeof(kernel_conf_file), path, utsname.release);
+        fp = fopen(path, "r");
+        if (fp != NULL) {
+            ucs_trace("reading kernel config from %s", path);
+            break;
+        }
+        ucs_trace("could not open kernel conf file %s error: %m", path);
+    }
+
+    if (fp != NULL) {
+        while (fgets(buf, sizeof(buf), fp) != NULL) {
+            if (strstr(buf, kernel_opt1) != NULL) {
+                found_opt1 = 1;
+            }
+            if (strstr(buf, kernel_opt2) != NULL) {
+                found_opt2 = 1;
+            }
+            if (found_opt1 && found_opt2) {
+                dmabuf_supported = 1;
+                break;
+            }
+        }
+
+        fclose(fp);
+        return dmabuf_supported;
+    }
+
+    ucs_trace("no kernel conf file found, trying /proc/kallsyms fallback");
+    fp = fopen("/proc/kallsyms", "r");
+
     if (fp == NULL) {
-        ucs_trace("could not open kernel conf file %s error: %m",
-                  kernel_conf_file);
+        ucs_trace("no kernel conf or kallsyms file found");
         goto out;
     }
 
     while (fgets(buf, sizeof(buf), fp) != NULL) {
-        if (strstr(buf, kernel_opt1) != NULL) {
-            found_opt1 = 1;
-        }
-        if (strstr(buf, kernel_opt2) != NULL) {
-            found_opt2 = 1;
-        }
+        if (!found_opt1 && strstr(buf, "dma_buf_move_notify") != NULL)
+          found_opt1 = 1;
+        if (!found_opt2 && strstr(buf, "pci_p2pdma") != NULL)
+          found_opt2 = 1;
         if (found_opt1 && found_opt2) {
             dmabuf_supported = 1;
             break;
         }
     }
     fclose(fp);
+    
 #endif
 out:
     return dmabuf_supported;