提交 560bafe5 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix the warning log.

上级 342d5018
import copy import copy
import theano
import numpy import numpy
import logging
from six.moves import xrange from six.moves import xrange
try: try:
...@@ -8,6 +8,7 @@ try: ...@@ -8,6 +8,7 @@ try:
except ImportError: except ImportError:
pass pass
import theano
from theano import tensor, scalar, gof from theano import tensor, scalar, gof
from theano.compile import optdb from theano.compile import optdb
from theano.compile.ops import shape_i from theano.compile.ops import shape_i
...@@ -41,6 +42,8 @@ from .subtensor import (GpuIncSubtensor, GpuSubtensor, ...@@ -41,6 +42,8 @@ from .subtensor import (GpuIncSubtensor, GpuSubtensor,
GpuAdvancedIncSubtensor1_dev20) GpuAdvancedIncSubtensor1_dev20)
from .opt_util import alpha_merge, output_merge from .opt_util import alpha_merge, output_merge
_logger = logging.getLogger("theano.sandbox.gpuarray.opt")
gpu_optimizer = EquilibriumDB() gpu_optimizer = EquilibriumDB()
gpu_cut_copies = EquilibriumDB() gpu_cut_copies = EquilibriumDB()
...@@ -611,7 +614,9 @@ def local_gpua_gemm(node): ...@@ -611,7 +614,9 @@ def local_gpua_gemm(node):
def local_gpua_hgemm(node): def local_gpua_hgemm(node):
from theano.sandbox.cuda import nvcc_compiler from theano.sandbox.cuda import nvcc_compiler
if nvcc_compiler.nvcc_version < '7.5': if nvcc_compiler.nvcc_version < '7.5':
log.warning("Not performing dot of float16 on the GPU since cuda 7.5 is not available. Updating could speed up your code.") _logger.warning("Not performing dot of float16 on the GPU since "
"cuda 7.5 is not available. Updating could speed up "
"your code.")
return return
A = node.inputs[0] A = node.inputs[0]
B = node.inputs[1] B = node.inputs[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论