From e91fc2ddaa74dd2c46ce93c9e92020d66c037c8e Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Wed, 24 Feb 2016 16:58:32 +0800
Subject: add CuContext/MContext

---
 nerv/nerv | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'nerv/nerv')

diff --git a/nerv/nerv b/nerv/nerv
index 7571659..4dd448c 100644
--- a/nerv/nerv
+++ b/nerv/nerv
@@ -1,6 +1,24 @@
 #! /usr/bin/env luajit
 require 'nerv'
-print("Greetings")
+nerv.printf("*** NERV: A Lua-based toolkit for high-performance deep learning (alpha) ***\n")
+nerv.info("automatically initialize a default CuContext...")
+nerv.CuMatrix._default_context = nerv.CuContext()
+nerv.info("the default CuContext is ok")
+
+nerv.info("automatically initialize a default MContext...")
+nerv.MMatrix._default_context = nerv.MContext()
+nerv.info("the default MContext is ok")
+
+-- only for backward compatibilty, will be removed in the future
+local function _add_profile_method(cls)
+    local c = cls._default_context
+    cls.print_profile = function () c:print_profile() end
+    cls.clear_profile = function () c:clear_profile() end
+end
+_add_profile_method(nerv.CuMatrix)
+_add_profile_method(nerv.MMatrix)
+
+
 if #arg < 1 then
     return
 end
@@ -11,3 +29,5 @@ for i = 2, #arg do
 end
 arg = script_arg
 dofile(script)
+nerv.CuMatrix.print_profile()
+nerv.MMatrix.print_profile()
-- 
cgit v1.2.3