Fix initial-release nvcc v10.1 issue with thread_local.h (#579)

dmlc · Nov 5, 2019 · ca9f932 · ca9f932
1 parent 05fb5ad
commit ca9f932
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/include/dmlc/thread_local.h b/include/dmlc/thread_local.h
@@ -43,7 +43,9 @@ class ThreadLocalStore {
     static MX_THREAD_LOCAL T* ptr = nullptr;
     if (ptr == nullptr) {
       ptr = new T();
-      Singleton()->RegisterDelete(ptr);
+      // Syntactic work-around for the nvcc of the initial cuda v10.1 release,
+      // which fails to compile 'Singleton()->' below. Fixed in v10.1 update 1.
+      (*Singleton()).RegisterDelete(ptr);
     }
     return ptr;
 #endif