提交 5ce32c02 authored 作者: James Bergstra's avatar James Bergstra

cleaned up logging in cuda.__init__

上级 6f92ed1e
...@@ -6,13 +6,14 @@ import logging, copy ...@@ -6,13 +6,14 @@ import logging, copy
_logger_name = 'theano.sandbox.cuda' _logger_name = 'theano.sandbox.cuda'
_logger = logging.getLogger(_logger_name) _logger = logging.getLogger(_logger_name)
_logger.setLevel(logging.WARNING) _logger.setLevel(logging.WARNING)
_logger.addHandler(logging.StreamHandler()) def error(*msg):
_logger.warning('ERROR (%s): '% ( _logger_name, ' '.join(str(m) for m in msg)))
def warning(*msg): def warning(*msg):
_logger.warning(_logger_name+'WARNING: '+' '.join(str(m) for m in msg)) _logger.warning('WARNING (%s): '% ( _logger_name, ' '.join(str(m) for m in msg)))
def info(*msg): def info(*msg):
_logger.info(_logger_name+'INFO: '+' '.join(str(m) for m in msg)) _logger.warning('INFO (%s): '% ( _logger_name, ' '.join(str(m) for m in msg)))
def debug(*msg): def debug(*msg):
_logger.debug(_logger_name+'DEBUG: '+' '.join(str(m) for m in msg)) _logger.warning('DEBUG (%s): '% ( _logger_name, ' '.join(str(m) for m in msg)))
# Compile cuda_ndarray.cu # Compile cuda_ndarray.cu
...@@ -80,7 +81,7 @@ try: ...@@ -80,7 +81,7 @@ try:
from cuda_ndarray.cuda_ndarray import * from cuda_ndarray.cuda_ndarray import *
except Exception, e: except Exception, e:
_logger.error( "Failed to compile cuda_ndarray.cu: %s" % str(e)) error( "Failed to compile cuda_ndarray.cu: %s" % str(e))
set_cuda_disabled() set_cuda_disabled()
if enable_cuda: if enable_cuda:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论