[Polly][GPGPU] Fixed undefined reference for CUDA's managed memory in Runtime library.

llvm-svn: 311848

1 files changed, 24 insertions, 2 deletions

diff --git a/polly/tools/GPURuntime/GPUJIT.c b/polly/tools/GPURuntime/GPUJIT.c
index 935d682380a..f68839ae694 100644
--- a/polly/tools/GPURuntime/GPUJIT.c
+++ b/polly/tools/GPURuntime/GPUJIT.c
@@ -1453,7 +1453,7 @@ int isManagedPtr(void *mem) {
   return 0;
 }
 
-void polly_freeManaged(void *mem) {
+void freeManagedCUDA(void *mem) {
   dump_function();
 
   // In a real-world program this was used (COSMO), there were more `free`
@@ -1475,7 +1475,7 @@ void polly_freeManaged(void *mem) {
   }
 }
 
-void *polly_mallocManaged(size_t size) {
+void *mallocManagedCUDA(size_t size) {
   // Note: [Size 0 allocations]
   // Sometimes, some runtime computation of size could create a size of 0
   // for an allocation. In these cases, we do not wish to fail.
@@ -1812,6 +1812,28 @@ void polly_freeContext(PollyGPUContext *Context) {
   }
 }
 
+void polly_freeManaged(void *mem) {
+  dump_function();
+
+#ifdef HAS_LIBCUDART
+  freeManagedCUDA(mem);
+#else
+  fprintf(stderr, "No CUDA Runtime. Managed memory only supported by CUDA\n");
+  exit(-1);
+#endif
+}
+
+void *polly_mallocManaged(size_t size) {
+  dump_function();
+
+#ifdef HAS_LIBCUDART
+  return mallocManagedCUDA(size);
+#else
+  fprintf(stderr, "No CUDA Runtime. Managed memory only supported by CUDA\n");
+  exit(-1);
+#endif
+}
+
 /* Initialize GPUJIT with CUDA as runtime library. */
 PollyGPUContext *polly_initContextCUDA() {
 #ifdef HAS_LIBCUDART