1 files changed, 79 insertions, 0 deletions
diff --git a/src/CuBaseLib/cudevice.h b/src/CuBaseLib/cudevice.h
new file mode 100644
index 0000000..c5eeb7b
--- /dev/null
+++ b/src/CuBaseLib/cudevice.h
@@ -0,0 +1,79 @@
+#ifndef _CUDEVICE_H_
+#define _CUDEVICE_H_
+
+#include <map>
+#include <string>
+#include <iostream>
+
+namespace TNet {
+
+  /**
+   * Singleton object which represents CUDA device
+   * responsible for CUBLAS initilalisation
+   * and memory block registration
+   */
+  class CuDevice 
+  {
+    // Singleton interface...
+    private:
+      CuDevice();
+      CuDevice(CuDevice&);
+      CuDevice& operator=(CuDevice&);
+
+    public:
+      ~CuDevice();
+      static CuDevice& Instantiate()
+      { return msDevice; }
+
+    private:
+      static CuDevice msDevice;
+
+
+    /**********************************/
+    // Instance interface
+    public:
+
+      void SelectGPU(int gpu_id);
+     
+      /// Check if the CUDA device is in the system      
+      bool IsPresent()
+      { return mIsPresent; }
+
+      void Verbose(bool verbose)
+      { mVerbose = verbose; }
+
+      /// Sum the IO time
+      void AccuProfile(const std::string& key,double time) 
+      { 
+        if(mProfileMap.find(key) == mProfileMap.end()) {
+          mProfileMap[key] = 0.0;
+        }
+        mProfileMap[key] += time;
+      }
+
+      void PrintProfile()
+      { 
+        std::cout << "[cudevice profile]\n";
+        std::map<std::string, double>::iterator it;
+        for(it = mProfileMap.begin(); it != mProfileMap.end(); ++it) {
+          std::cout << it->first << "\t" << it->second << "s\n";
+        }
+      }
+
+      void ResetProfile()
+      { mProfileMap.clear(); }
+
+      std::string GetFreeMemory();
+
+
+    private:
+      std::map<std::string, double> mProfileMap;
+      bool mIsPresent;
+      bool mVerbose;
+  }; //class CuDevice
+
+
+}
+
+
+#endif